MCPcopy Create free account

hub / github.com/HVision-NKU/DepthAnythingAC / functions

Functions579 in github.com/HVision-NKU/DepthAnythingAC

⨍ Functions 579◇ Types & classes 98
↓ 2 callersFunctionsetup_distributed
AdaHessian Optimizer Lifted from https://github.com/BIGBALLON/distribuuuu/blob/master/distribuuuu/utils.py Originally licensed MIT, Copyright
util/dist_helper.py:8
↓ 2 callersMethodsinkhorn_knopp_teacher
(self, teacher_output, teacher_temp, n_iterations=3)
torchhub/facebookresearch_dinov2_main/dinov2/loss/dino_clstoken_loss.py:36
↓ 2 callersMethodsoftmax_center_teacher
(self, teacher_output, teacher_temp)
torchhub/facebookresearch_dinov2_main/dinov2/loss/dino_clstoken_loss.py:30
↓ 2 callersMethodsynchronize_between_processes
(self)
torchhub/facebookresearch_dinov2_main/dinov2/logging/helpers.py:47
↓ 2 callersFunctiontrain_for_C
(*, C, max_iter, train_features, train_labels, dtype=torch.float64, device=_CPU_DEVICE)
torchhub/facebookresearch_dinov2_main/dinov2/eval/log_regression.py:154
↓ 2 callersFunctiontransform
(sample)
torchhub/facebookresearch_dinov2_main/dinov2/data/loaders.py:34
↓ 2 callersMethodupdate
(self, value, num=1)
torchhub/facebookresearch_dinov2_main/dinov2/logging/helpers.py:147
↓ 2 callersMethodupdate_center
(self, teacher_output)
torchhub/facebookresearch_dinov2_main/dinov2/loss/dino_clstoken_loss.py:78
↓ 1 callersMethod__init__
(self, model)
torchhub/facebookresearch_dinov2_main/dinov2/eval/utils.py:23
↓ 1 callersMethod__init__
(self, train_features, train_labels, nb_knn, T, device, num_classes=1000)
torchhub/facebookresearch_dinov2_main/dinov2/eval/knn.py:110
↓ 1 callersMethod__init__
( self, in_features: int, hidden_features: Optional[int] = None, out_features:
torchhub/facebookresearch_dinov2_main/dinov2/layers/swiglu_ffn.py:14
↓ 1 callersMethod__init__
Init. Args: features (int): number of features
depth_anything/util/blocksv1.py:41
↓ 1 callersMethod__init__
Init. Args: features (int): number of features
depth_anything/util/blocksv2.py:33
↓ 1 callersMethod__init__
( self, in_features: int, hidden_features: Optional[int] = None, out_features:
depth_anything/dinov2_layers/swiglu_ffn.py:14
↓ 1 callersFunction_build_mlp
(nlayers, in_dim, bottleneck_dim, hidden_dim=None, use_bn=False, bias=True)
torchhub/facebookresearch_dinov2_main/dinov2/layers/dino_head.py:45
↓ 1 callersFunction_collect_env_vars
()
torchhub/facebookresearch_dinov2_main/dinov2/distributed/__init__.py:121
↓ 1 callersFunction_configure_logger
Configure a logger. Adapted from Detectron2. Args: name: The name of the logger to configure. level: The logging level
torchhub/facebookresearch_dinov2_main/dinov2/logging/__init__.py:19
↓ 1 callersMethod_dump_class_ids
(self, *args, **kwargs)
torchhub/facebookresearch_dinov2_main/dinov2/data/datasets/image_net_22k.py:281
↓ 1 callersMethod_dump_class_ids_and_names
(self)
torchhub/facebookresearch_dinov2_main/dinov2/data/datasets/image_net.py:253
↓ 1 callersMethod_dump_entries
(self, *args, **kwargs)
torchhub/facebookresearch_dinov2_main/dinov2/data/datasets/image_net_22k.py:243
↓ 1 callersMethod_dump_entries
(self)
torchhub/facebookresearch_dinov2_main/dinov2/data/datasets/image_net.py:187
↓ 1 callersMethod_dump_extra
(self, *args, **kwargs)
torchhub/facebookresearch_dinov2_main/dinov2/data/datasets/image_net_22k.py:298
↓ 1 callersMethod_find_class_ids
(self, path: str)
torchhub/facebookresearch_dinov2_main/dinov2/data/datasets/image_net_22k.py:130
↓ 1 callersMethod_gather_all_knn_for_rank
(self, topk_sims, neighbors_labels, target_rank)
torchhub/facebookresearch_dinov2_main/dinov2/eval/knn.py:145
↓ 1 callersFunction_generate_randperm_indices
Generate the indices of a random permutation.
torchhub/facebookresearch_dinov2_main/dinov2/data/samplers.py:64
↓ 1 callersFunction_get_available_port
()
torchhub/facebookresearch_dinov2_main/dinov2/distributed/__init__.py:102
↓ 1 callersMethod_get_class_ids
(self)
torchhub/facebookresearch_dinov2_main/dinov2/data/datasets/image_net.py:112
↓ 1 callersMethod_get_class_names
(self)
torchhub/facebookresearch_dinov2_main/dinov2/data/datasets/image_net.py:120
↓ 1 callersMethod_get_intermediate_layers_chunked
(self, x, n=1)
torchhub/facebookresearch_dinov2_main/vision_transformer.py:283
↓ 1 callersMethod_get_intermediate_layers_chunked
(self, x, n=1)
torchhub/facebookresearch_dinov2_main/dinov2/models/vision_transformer.py:250
↓ 1 callersMethod_get_intermediate_layers_chunked
(self, x, n=1)
depth_anything/dinov2.py:283
↓ 1 callersMethod_get_intermediate_layers_not_chunked
(self, x, n=1)
torchhub/facebookresearch_dinov2_main/vision_transformer.py:271
↓ 1 callersMethod_get_intermediate_layers_not_chunked
(self, x, n=1)
torchhub/facebookresearch_dinov2_main/dinov2/models/vision_transformer.py:238
↓ 1 callersMethod_get_intermediate_layers_not_chunked
(self, x, n=1)
depth_anything/dinov2.py:271
↓ 1 callersFunction_get_master_port
(seed: int = 0)
torchhub/facebookresearch_dinov2_main/dinov2/distributed/__init__.py:91
↓ 1 callersFunction_get_numpy_dtype
(size: int)
torchhub/facebookresearch_dinov2_main/dinov2/data/samplers.py:56
↓ 1 callersMethod_get_pad
(self, size)
torchhub/facebookresearch_dinov2_main/utils.py:28
↓ 1 callersFunction_get_tarball_path
(class_id: str)
torchhub/facebookresearch_dinov2_main/dinov2/data/datasets/image_net_22k.py:56
↓ 1 callersFunction_guess_cluster_type
()
torchhub/facebookresearch_dinov2_main/dinov2/utils/cluster.py:19
↓ 1 callersFunction_is_slurm_job_process
()
torchhub/facebookresearch_dinov2_main/dinov2/distributed/__init__.py:125
↓ 1 callersMethod_iterator
(self)
torchhub/facebookresearch_dinov2_main/dinov2/data/samplers.py:105
↓ 1 callersMethod_iterator
(self)
torchhub/facebookresearch_dinov2_main/dinov2/data/samplers.py:202
↓ 1 callersMethod_load_entries_class_ids
(self, root: Optional[str] = None)
torchhub/facebookresearch_dinov2_main/dinov2/data/datasets/image_net_22k.py:142
↓ 1 callersMethod_load_labels
(self, labels_path: str)
torchhub/facebookresearch_dinov2_main/dinov2/data/datasets/image_net.py:172
↓ 1 callersFunction_make_mmap_tarball
(tarballs_root: str, mmap_cache_size: int)
torchhub/facebookresearch_dinov2_main/dinov2/data/datasets/image_net_22k.py:60
↓ 1 callersFunction_make_sampler
( *, dataset, type: Optional[SamplerType] = None, shuffle: bool = False, seed: int = 0,
torchhub/facebookresearch_dinov2_main/dinov2/data/loaders.py:101
↓ 1 callersFunction_make_scratch
(in_shape, out_shape, groups=1, expand=False)
depth_anything/util/blocksv1.py:4
↓ 1 callersFunction_make_scratch
(in_shape, out_shape, groups=1, expand=False)
depth_anything/util/blocksv2.py:4
↓ 1 callersFunction_make_seed
(seed: int, start: int, iter_count: int)
torchhub/facebookresearch_dinov2_main/dinov2/data/samplers.py:161
↓ 1 callersMethod_mask
(self, mask, max_mask_patches)
torchhub/facebookresearch_dinov2_main/dinov2/data/masking.py:50
↓ 1 callersFunction_parse_dataset_str
(dataset_str: str)
torchhub/facebookresearch_dinov2_main/dinov2/data/loaders.py:45
↓ 1 callersFunction_parse_slurm_node_list
(s: str)
torchhub/facebookresearch_dinov2_main/dinov2/distributed/__init__.py:129
↓ 1 callersFunction_restrict_print_to_main_process
This function disables printing when not in the main process
torchhub/facebookresearch_dinov2_main/dinov2/distributed/__init__.py:75
↓ 1 callersMethod_set_from_local
(self)
torchhub/facebookresearch_dinov2_main/dinov2/distributed/__init__.py:211
↓ 1 callersMethod_set_from_preset_env
(self)
torchhub/facebookresearch_dinov2_main/dinov2/distributed/__init__.py:199
↓ 1 callersMethod_set_from_slurm_env
(self)
torchhub/facebookresearch_dinov2_main/dinov2/distributed/__init__.py:182
↓ 1 callersMethod_setup_args
(self)
torchhub/facebookresearch_dinov2_main/dinov2/run/eval/linear.py:36
↓ 1 callersMethod_setup_args
(self)
torchhub/facebookresearch_dinov2_main/dinov2/run/eval/log_regression.py:36
↓ 1 callersMethod_setup_args
(self)
torchhub/facebookresearch_dinov2_main/dinov2/run/eval/knn.py:36
↓ 1 callersMethod_setup_args
(self)
torchhub/facebookresearch_dinov2_main/dinov2/run/train/train.py:36
↓ 1 callersMethod_shuffled_iterator
(self)
torchhub/facebookresearch_dinov2_main/dinov2/data/samplers.py:112
↓ 1 callersMethod_shuffled_iterator
(self)
torchhub/facebookresearch_dinov2_main/dinov2/data/samplers.py:209
↓ 1 callersMethod_similarity_for_rank
(self, features_rank, source_rank)
torchhub/facebookresearch_dinov2_main/dinov2/eval/knn.py:130
↓ 1 callersFunctionadd_residual
(x, brange, residual, residual_scale_factor, scaling_vector=None)
torchhub/facebookresearch_dinov2_main/dinov2/layers/block.py:142
↓ 1 callersFunctionadd_residual
(x, brange, residual, residual_scale_factor, scaling_vector=None)
depth_anything/dinov2_layers/block.py:142
↓ 1 callersMethodapply_center_update
(self)
torchhub/facebookresearch_dinov2_main/dinov2/loss/ibot_patch_loss.py:142
↓ 1 callersMethodapply_center_update
(self)
torchhub/facebookresearch_dinov2_main/dinov2/loss/dino_clstoken_loss.py:90
↓ 1 callersFunctionapply_optim_scheduler
(optimizer, lr, wd, last_layer_lr)
torchhub/facebookresearch_dinov2_main/dinov2/train/train.py:114
↓ 1 callersFunctionapply_scaling_rules_to_cfg
(cfg)
torchhub/facebookresearch_dinov2_main/dinov2/utils/config.py:22
↓ 1 callersFunctionas_torch_dtype
(dtype: TypeSpec)
torchhub/facebookresearch_dinov2_main/dinov2/utils/dtype.py:32
↓ 1 callersMethodbackprop_loss
(self, loss)
torchhub/facebookresearch_dinov2_main/dinov2/train/ssl_meta_arch.py:129
↓ 1 callersFunctionbuild_model
(args, only_teacher=False, img_size=224)
torchhub/facebookresearch_dinov2_main/dinov2/models/__init__.py:15
↓ 1 callersFunctionbuild_model_for_eval
(config, pretrained_weights)
torchhub/facebookresearch_dinov2_main/dinov2/eval/setup.py:63
↓ 1 callersFunctionbuild_optimizer
(cfg, params_groups)
torchhub/facebookresearch_dinov2_main/dinov2/train/train.py:62
↓ 1 callersFunctionbuild_schedulers
(cfg)
torchhub/facebookresearch_dinov2_main/dinov2/train/train.py:66
↓ 1 callersFunctionbuild_topk_imagenet_real_accuracy_metric
(num_classes: int, ks: tuple = (1, 5))
torchhub/facebookresearch_dinov2_main/dinov2/eval/metrics.py:67
↓ 1 callersMethodcompute
(self)
torchhub/facebookresearch_dinov2_main/dinov2/eval/metrics.py:112
↓ 1 callersFunctioncompute_errors
Compute metrics for 'pred' compared to 'gt' Args: gt (numpy.ndarray): Ground truth values pred (numpy.ndarray): Predicted values
evaluate_depth.py:72
↓ 1 callersFunctioncompute_metrics
Compute depth estimation metrics for predicted depth maps. This function processes predictions and ground truth depth maps by: 1. Ap
evaluate_depth.py:116
↓ 1 callersMethodcompute_neighbors
(self, features_rank)
torchhub/facebookresearch_dinov2_main/dinov2/eval/knn.py:163
↓ 1 callersMethodcompute_scale_and_shift
Compute optimal scale and shift parameters to align prediction with target. This method solves a least squares problem to fi
evaluate_depth.py:182
↓ 1 callersFunctioncount_params
(model)
util/utils.py:6
↓ 1 callersFunctioncreate_class_indices_mapping
(labels)
torchhub/facebookresearch_dinov2_main/dinov2/eval/knn.py:235
↓ 1 callersFunctioncreate_contrast
Create corruptions: 'Contrast'.
perturbation/perturbation.py:146
↓ 1 callersFunctioncreate_fog
Create corruptions: 'Fog'.
perturbation/perturbation.py:59
↓ 1 callersFunctioncreate_frost
Create corruptions: 'Frost'.
perturbation/perturbation.py:76
↓ 1 callersFunctioncreate_module_dict
(*, module, n_per_class_list, n_tries, nb_knn, train_features, train_labels)
torchhub/facebookresearch_dinov2_main/dinov2/eval/knn.py:199
↓ 1 callersFunctioncreate_snow
Create corruptions: 'Snow'.
perturbation/perturbation.py:93
↓ 1 callersFunctioncrop
(sample, size)
dataset/transform.py:221
↓ 1 callersFunctiondefault_setup
(args)
torchhub/facebookresearch_dinov2_main/dinov2/utils/config.py:50
↓ 1 callersFunctiondenormalize
Reverses the imagenet normalization applied to the input. Args: x (torch.Tensor - shape(N,3,H,W)): input tensor Returns: tor
evaluate_depth.py:36
↓ 1 callersFunctiondo_train
(cfg, model, resume=False)
torchhub/facebookresearch_dinov2_main/dinov2/train/train.py:135
↓ 1 callersFunctiondrop_path
(x, drop_prob: float = 0.0, training: bool = False)
torchhub/facebookresearch_dinov2_main/dinov2/layers/drop_path.py:15
↓ 1 callersFunctiondrop_path
(x, drop_prob: float = 0.0, training: bool = False)
depth_anything/dinov2_layers/drop_path.py:15
↓ 1 callersMethoddump_in_output_file
(self, iteration, iter_time, data_time)
torchhub/facebookresearch_dinov2_main/dinov2/logging/helpers.py:54
↓ 1 callersFunctioneval_knn
( model, train_dataset, val_dataset, accuracy_averaging, nb_knn, temperature, batc
torchhub/facebookresearch_dinov2_main/dinov2/eval/knn.py:246
↓ 1 callersFunctioneval_knn_with_model
( model, output_dir, train_dataset_str="ImageNet:split=TRAIN", val_dataset_str="ImageNet:split
torchhub/facebookresearch_dinov2_main/dinov2/eval/knn.py:319
↓ 1 callersFunctioneval_linear
( *, feature_model, linear_classifiers, train_data_loader, val_data_loader, metrics_fi
torchhub/facebookresearch_dinov2_main/dinov2/eval/linear.py:314
↓ 1 callersFunctioneval_log_regression
Implements the "standard" process for log regression evaluation: The value of C is chosen by training on train_dataset and evaluating on
torchhub/facebookresearch_dinov2_main/dinov2/eval/log_regression.py:253
↓ 1 callersFunctioneval_log_regression_with_model
( model, train_dataset_str="ImageNet:split=TRAIN", val_dataset_str="ImageNet:split=VAL", finet
torchhub/facebookresearch_dinov2_main/dinov2/eval/log_regression.py:363
← previousnext →101–200 of 579, ranked by callers