MCPcopy Create free account

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

Functions579 in github.com/HVision-NKU/DepthAnythingAC

⨍ Functions 579◇ Types & classes 98
↓ 1 callersMethodexport
(self, *, overwrite: bool)
torchhub/facebookresearch_dinov2_main/dinov2/distributed/__init__.py:220
↓ 1 callersFunctionextract_features_with_dataloader
(model, data_loader, sample_count, gather_on_cpu=False)
torchhub/facebookresearch_dinov2_main/dinov2/eval/utils.py:114
↓ 1 callersFunctionfilter_train
(mapping, n_per_class, seed)
torchhub/facebookresearch_dinov2_main/dinov2/eval/knn.py:226
↓ 1 callersMethodfit
(self, train_features, train_labels)
torchhub/facebookresearch_dinov2_main/dinov2/eval/log_regression.py:138
↓ 1 callersMethodforward
(self, x: Tensor)
torchhub/facebookresearch_dinov2_main/dinov2/layers/block.py:82
↓ 1 callersMethodforward
(self, x: Tensor)
torchhub/facebookresearch_dinov2_main/dinov2/layers/attention.py:49
↓ 1 callersMethodforward
(self, x: Tensor)
depth_anything/dinov2_layers/block.py:82
↓ 1 callersMethodforward
(self, x: Tensor)
depth_anything/dinov2_layers/attention.py:49
↓ 1 callersMethodforward_backward
(self, images, teacher_temp)
torchhub/facebookresearch_dinov2_main/dinov2/train/ssl_meta_arch.py:135
↓ 1 callersMethodforward_features
(self, x, masks=None)
torchhub/facebookresearch_dinov2_main/vision_transformer.py:253
↓ 1 callersMethodforward_features
(self, x, masks=None)
torchhub/facebookresearch_dinov2_main/dinov2/models/vision_transformer.py:221
↓ 1 callersMethodforward_features
(self, x, masks=None)
depth_anything/dinov2.py:253
↓ 1 callersMethodforward_features_list
(self, x_list, masks_list)
torchhub/facebookresearch_dinov2_main/vision_transformer.py:233
↓ 1 callersMethodforward_features_list
(self, x_list, masks_list)
torchhub/facebookresearch_dinov2_main/dinov2/models/vision_transformer.py:202
↓ 1 callersMethodforward_features_list
(self, x_list, masks_list)
depth_anything/dinov2.py:233
↓ 1 callersMethodforward_masked
( self, student_patch_tokens_masked, teacher_patch_tokens_masked, student_mask
torchhub/facebookresearch_dinov2_main/dinov2/loss/ibot_patch_loss.py:106
↓ 1 callersMethodforward_nested
x_list contains a list of tensors to nest together and run
torchhub/facebookresearch_dinov2_main/dinov2/layers/block.py:205
↓ 1 callersMethodforward_nested
x_list contains a list of tensors to nest together and run
depth_anything/dinov2_layers/block.py:205
↓ 1 callersFunctionfree_if_fsdp
(x)
torchhub/facebookresearch_dinov2_main/dinov2/fsdp/__init__.py:66
↓ 1 callersMethodfsdp_synchronize_streams
(self)
torchhub/facebookresearch_dinov2_main/dinov2/train/ssl_meta_arch.py:351
↓ 1 callersFunctionfuse_params_groups
(all_params_groups, keys=("lr_multiplier", "wd_multiplier", "is_last_layer"))
torchhub/facebookresearch_dinov2_main/dinov2/utils/param_groups.py:83
↓ 1 callersMethodgenerate_depth_decay
generate 2d decay mask, the result is (HW)*(HW) H, W are the numbers of patches at each column and row
depth_anything/util/priorgenerate.py:15
↓ 1 callersMethodgenerate_pos_decay
generate 2d decay mask, the result is (HW)*(HW) H, W are the numbers of patches at each column and row
depth_anything/util/priorgenerate.py:27
↓ 1 callersFunctionget_args_parser
( description: Optional[str] = None, parents: Optional[List[argparse.ArgumentParser]] = None, add_
torchhub/facebookresearch_dinov2_main/dinov2/eval/linear.py:34
↓ 1 callersFunctionget_args_parser
( description: Optional[str] = None, parents: Optional[List[argparse.ArgumentParser]] = None, add_
torchhub/facebookresearch_dinov2_main/dinov2/eval/log_regression.py:39
↓ 1 callersFunctionget_args_parser
( description: Optional[str] = None, parents: Optional[List[argparse.ArgumentParser]] = None, add_
torchhub/facebookresearch_dinov2_main/dinov2/eval/knn.py:30
↓ 1 callersFunctionget_args_parser
(add_help: bool = True)
torchhub/facebookresearch_dinov2_main/dinov2/train/train.py:31
↓ 1 callersFunctionget_augment_list
()
dataset/transform.py:487
↓ 1 callersFunctionget_autocast_dtype
(config)
torchhub/facebookresearch_dinov2_main/dinov2/eval/setup.py:53
↓ 1 callersFunctionget_branges_scales
(x, sample_drop_ratio=0.0)
torchhub/facebookresearch_dinov2_main/dinov2/layers/block.py:134
↓ 1 callersFunctionget_branges_scales
(x, sample_drop_ratio=0.0)
depth_anything/dinov2_layers/block.py:134
↓ 1 callersFunctionget_cfg_from_args
(args)
torchhub/facebookresearch_dinov2_main/dinov2/utils/config.py:41
↓ 1 callersFunctionget_checkpoint_path
(cluster_type: Optional[ClusterType] = None)
torchhub/facebookresearch_dinov2_main/dinov2/utils/cluster.py:39
↓ 1 callersMethodget_class_id
(self, index: int)
torchhub/facebookresearch_dinov2_main/dinov2/data/datasets/image_net.py:157
↓ 1 callersMethodget_image_data
(self, index: int)
torchhub/facebookresearch_dinov2_main/dinov2/data/datasets/extended.py:18
↓ 1 callersMethodget_image_relpath
(self, actual_index: int, class_id: Optional[str] = None)
torchhub/facebookresearch_dinov2_main/dinov2/data/datasets/image_net.py:39
↓ 1 callersMethodget_intermediate_layers
( self, x: torch.Tensor, n: Union[int, Sequence] = 1, # Layers or n last layers to ta
torchhub/facebookresearch_dinov2_main/dinov2/models/vision_transformer.py:264
↓ 1 callersMethodget_maybe_fused_params_for_submodel
(self, m)
torchhub/facebookresearch_dinov2_main/dinov2/train/ssl_meta_arch.py:374
↓ 1 callersFunctionget_package_version
()
torchhub/facebookresearch_dinov2_main/setup.py:43
↓ 1 callersMethodget_params_groups
(self)
torchhub/facebookresearch_dinov2_main/dinov2/train/ssl_meta_arch.py:387
↓ 1 callersFunctionget_params_groups_with_decay
(model, lr_decay_rate=1.0, patch_embed_lr_mult=1.0)
torchhub/facebookresearch_dinov2_main/dinov2/utils/param_groups.py:42
↓ 1 callersMethodget_shape
(self)
torchhub/facebookresearch_dinov2_main/dinov2/data/masking.py:47
↓ 1 callersFunctionget_shared_folder
()
torchhub/facebookresearch_dinov2_main/dinov2/run/submit.py:84
↓ 1 callersMethodget_size
(self, width, height)
dataset/transform.py:68
↓ 1 callersMethodget_size
(self, width, height)
depth_anything/util/transform.py:111
↓ 1 callersFunctionget_slurm_executor_parameters
( nodes: int, num_gpus_per_node: int, cluster_type: Optional[ClusterType] = None, **kwargs )
torchhub/facebookresearch_dinov2_main/dinov2/utils/cluster.py:75
↓ 1 callersMethodget_target
(self, index: int)
torchhub/facebookresearch_dinov2_main/dinov2/data/datasets/extended.py:21
↓ 1 callersMethodget_targets
(self)
torchhub/facebookresearch_dinov2_main/dinov2/data/datasets/image_net.py:153
↓ 1 callersFunctionget_user_checkpoint_path
(cluster_type: Optional[ClusterType] = None)
torchhub/facebookresearch_dinov2_main/dinov2/utils/cluster.py:52
↓ 1 callersMethodget_value
(self)
evaluate_depth.py:32
↓ 1 callersFunctionget_vit_lr_decay_rate
Calculate lr decay rate for different ViT blocks. Args: name (string): parameter name. lr_decay_rate (float): base lr decay r
torchhub/facebookresearch_dinov2_main/dinov2/utils/param_groups.py:14
↓ 1 callersFunctionhas_batchnorms
(model)
torchhub/facebookresearch_dinov2_main/dinov2/utils/utils.py:91
↓ 1 callersFunctionhas_ddp_wrapper
(m: nn.Module)
torchhub/facebookresearch_dinov2_main/dinov2/eval/linear.py:156
↓ 1 callersFunctionhflip
(sample, p=0.5)
dataset/transform.py:250
↓ 1 callersFunctionimadjust
(x, a, b, c, d, gamma=1)
perturbation/perturbation.py:8
↓ 1 callersFunctioninfer_batch_mixed
Batch process both images and videos in a directory Args: model: Trained depth estimation model input_dir (str): Input d
tools/infer.py:398
↓ 1 callersFunctioninfer_batch_videos
Batch process videos in a directory Args: model: Trained depth estimation model input_dir (str): Input directory contain
tools/infer.py:355
↓ 1 callersFunctioninit_seeds
(seed=0, cuda_deterministic=False)
full_train.py:85
↓ 1 callersMethodinit_weights
(self)
torchhub/facebookresearch_dinov2_main/vision_transformer.py:172
↓ 1 callersMethodinit_weights
(self)
torchhub/facebookresearch_dinov2_main/dinov2/models/vision_transformer.py:160
↓ 1 callersMethodinit_weights
(self)
depth_anything/dinov2.py:172
↓ 1 callersMethodinterpolate_pos_encoding
(self, x, w, h)
torchhub/facebookresearch_dinov2_main/vision_transformer.py:179
↓ 1 callersMethodinterpolate_pos_encoding
(self, x, w, h)
torchhub/facebookresearch_dinov2_main/dinov2/models/vision_transformer.py:165
↓ 1 callersMethodinterpolate_pos_encoding
(self, x, w, h)
depth_anything/dinov2.py:179
↓ 1 callersFunctionis_fsdp
(x)
torchhub/facebookresearch_dinov2_main/dinov2/fsdp/__init__.py:58
↓ 1 callersFunctionis_main_process
Returns: True if the current process is the main one.
torchhub/facebookresearch_dinov2_main/dinov2/distributed/__init__.py:67
↓ 1 callersFunctionis_sharded_fsdp
(x)
torchhub/facebookresearch_dinov2_main/dinov2/fsdp/__init__.py:62
↓ 1 callersFunctionload_model
Load trained depth estimation model
tools/infer.py:39
↓ 1 callersFunctionlossfunc
(t, s, temp)
torchhub/facebookresearch_dinov2_main/dinov2/loss/ibot_patch_loss.py:21
↓ 1 callersFunctionlow_light
(path, x, severity)
perturbation/perturbation.py:37
↓ 1 callersFunctionmain
()
full_train.py:98
↓ 1 callersFunctionmain
()
evaluate_depth.py:456
↓ 1 callersFunctionmain
()
tools/infer.py:474
↓ 1 callersFunctionmain
(args)
torchhub/facebookresearch_dinov2_main/dinov2/eval/linear.py:596
↓ 1 callersFunctionmain
(args)
torchhub/facebookresearch_dinov2_main/dinov2/eval/log_regression.py:424
↓ 1 callersFunctionmain
(args)
torchhub/facebookresearch_dinov2_main/dinov2/eval/knn.py:380
↓ 1 callersFunctionmain
(args)
torchhub/facebookresearch_dinov2_main/dinov2/train/train.py:298
↓ 1 callersFunctionmain
()
torchhub/facebookresearch_dinov2_main/dinov2/run/eval/linear.py:45
↓ 1 callersFunctionmain
()
torchhub/facebookresearch_dinov2_main/dinov2/run/eval/log_regression.py:45
↓ 1 callersFunctionmain
()
torchhub/facebookresearch_dinov2_main/dinov2/run/eval/knn.py:45
↓ 1 callersFunctionmain
()
torchhub/facebookresearch_dinov2_main/dinov2/run/train/train.py:45
↓ 1 callersFunctionmake_classification_train_transform
( *, crop_size: int = 224, interpolation=transforms.InterpolationMode.BICUBIC, hflip_prob: flo
torchhub/facebookresearch_dinov2_main/dinov2/data/transforms.py:56
↓ 1 callersFunctionnamed_apply
(fn: Callable, module: nn.Module, name="", depth_first=True, include_root=False)
torchhub/facebookresearch_dinov2_main/vision_transformer.py:26
↓ 1 callersFunctionnamed_apply
(fn: Callable, module: nn.Module, name="", depth_first=True, include_root=False)
torchhub/facebookresearch_dinov2_main/dinov2/models/vision_transformer.py:27
↓ 1 callersFunctionnamed_apply
(fn: Callable, module: nn.Module, name="", depth_first=True, include_root=False)
depth_anything/dinov2.py:26
↓ 1 callersFunctionnormalize_depth_only
(sample,dataset=None)
dataset/transform.py:284
↓ 1 callersMethodpairwise_NNs_inner
Pairwise nearest neighbors for L2-normalized vectors. Uses Torch rather than Faiss to remain on GPU.
torchhub/facebookresearch_dinov2_main/dinov2/loss/koleo_loss.py:26
↓ 1 callersMethodparse_image_relpath
(self, image_relpath: str)
torchhub/facebookresearch_dinov2_main/dinov2/data/datasets/image_net.py:47
↓ 1 callersFunctionperturbation
(path, image, cfg)
perturbation/perturbation.py:164
↓ 1 callersFunctionpoisson_gaussian_noise
(path,x, severity)
perturbation/perturbation.py:12
↓ 1 callersMethodprepare_for_distributed_training
(self)
torchhub/facebookresearch_dinov2_main/dinov2/train/ssl_meta_arch.py:393
↓ 1 callersFunctionread_depth_ETH3D
(filename)
dataset/semi_depth.py:31
↓ 1 callersFunctionread_depth_sintel
(filename)
dataset/semi_depth.py:16
↓ 1 callersMethodreduce_center_update
(self, teacher_patch_tokens)
torchhub/facebookresearch_dinov2_main/dinov2/loss/ibot_patch_loss.py:134
↓ 1 callersMethodreduce_center_update
(self, teacher_output)
torchhub/facebookresearch_dinov2_main/dinov2/loss/dino_clstoken_loss.py:82
↓ 1 callersMethodreset
(self)
util/utils.py:60
↓ 1 callersFunctionreshard_fsdp_model
(x)
torchhub/facebookresearch_dinov2_main/dinov2/fsdp/__init__.py:77
↓ 1 callersFunctionresize_certain
(sample, short_side)
dataset/transform.py:334
↓ 1 callersFunctionrun_eval_linear
( model, output_dir, train_dataset_str, val_dataset_str, batch_size, epochs, epoch
torchhub/facebookresearch_dinov2_main/dinov2/eval/linear.py:464
↓ 1 callersFunctionsave_depth_map
Save depth map
tools/infer.py:101
← previousnext →201–300 of 579, ranked by callers