MCPcopy Create free account

hub / github.com/IRMVLab/SemGauss-SLAM / functions

Functions506 in github.com/IRMVLab/SemGauss-SLAM

↓ 42 callersFunctionprint
(*args, **kwargs)
segmentation/facebookresearch_dinov2_main/dinov2/distributed/__init__.py:83
↓ 18 callersMethodsplit
(self)
segmentation/facebookresearch_dinov2_main/dinov2/data/datasets/image_net.py:79
↓ 18 callersMethodupdate
(self, **kwargs)
segmentation/facebookresearch_dinov2_main/dinov2/logging/helpers.py:27
↓ 17 callersMethodmax
(self)
segmentation/facebookresearch_dinov2_main/dinov2/logging/helpers.py:181
↓ 15 callersMethodload
(self, *args, **kwargs)
segmentation/facebookresearch_dinov2_main/dinov2/fsdp/__init__.py:115
↓ 12 callersMethodget
Gets the current evaluation result. Returns ------- metrics : tuple of float pixAcc and mIoU
utils/segmentationMetric.py:48
↓ 10 callersFunctionbuild_rotation
(q)
utils/slam_external.py:8
↓ 8 callersFunctionmake_dataset
Creates a dataset with the specified parameters. Args: dataset_str: A dataset string description (e.g. ImageNet:split=TRAIN).
segmentation/facebookresearch_dinov2_main/dinov2/data/loaders.py:68
↓ 8 callersMethodsave
Dump model and checkpointables to a file. Args: name (str): name of the file. kwargs (dict): extra arbitrary
segmentation/facebookresearch_dinov2_main/dinov2/fsdp/__init__.py:87
↓ 8 callersMethodupdate
Updates the internal evaluation result. Parameters ---------- labels : 'NumpyArray' or list of `NumpyArray` The l
utils/segmentationMetric.py:19
↓ 7 callersFunctiontransform_to_frame
Function to transform Isotropic Gaussians from world frame to camera frame. Args: params: dict of parameters time_idx: t
utils/slam_helpers.py:111
↓ 6 callersMethod_get_entries
(self)
segmentation/facebookresearch_dinov2_main/dinov2/data/datasets/image_net.py:106
↓ 6 callersMethodbackward
(ctx, grad_out_color, _, depth, grad_out_semantics)
diff-gaussian-rasterization-w-depth_sem_gauss/diff_gaussian_rasterization/__init__.py:98
↓ 5 callersFunction_make_dinov2_model
( *, arch_name: str = "vit_large", img_size: int = 518, patch_size: int = 14, init_values:
segmentation/facebookresearch_dinov2_main/hubconf.py:22
↓ 5 callersFunctionmake_data_loader
Creates a data loader with the specified parameters. Args: dataset: A dataset (third party, LaViDa or WebDataset). batch_siz
segmentation/facebookresearch_dinov2_main/dinov2/data/loaders.py:167
↓ 5 callersFunctionsetup_logging
Setup logging. Args: output: A file name or a directory to save log files. If None, log files will not be saved. If outp
segmentation/facebookresearch_dinov2_main/dinov2/logging/__init__.py:83
↓ 4 callersMethod_load_extra
(self, extra_path: str)
segmentation/facebookresearch_dinov2_main/dinov2/data/datasets/image_net.py:85
↓ 4 callersFunction_make_dinov2_linear_classifier
( *, arch_name: str = "vit_large", layers: int = 4, pretrained: bool = True, **kwargs, )
segmentation/facebookresearch_dinov2_main/hubconf.py:136
↓ 4 callersFunctiondecode_segmap
(image, nc=25)
utils/eval_utils.py:552
↓ 4 callersFunctionextract_features
(model, dataset, batch_size, num_workers, gather_on_cpu=False)
segmentation/facebookresearch_dinov2_main/dinov2/eval/utils.py:99
↓ 4 callersFunctionget_args_parser
( description: Optional[str] = None, parents: Optional[List[argparse.ArgumentParser]] = [], add_he
segmentation/facebookresearch_dinov2_main/dinov2/run/submit.py:25
↓ 4 callersFunctionget_loss
(params, curr_data, variables, iter_time_idx, loss_weights, use_sil_for_loss, sil_thres, use_l1,i
sem_gauss.py:180
↓ 4 callersFunctionis_enabled
Returns: True if distributed training is enabled
segmentation/facebookresearch_dinov2_main/dinov2/distributed/__init__.py:20
↓ 4 callersMethodlog_every
(self, iterable, print_freq, header=None, n_iterations=None, start_iteration=0)
segmentation/facebookresearch_dinov2_main/dinov2/logging/helpers.py:67
↓ 4 callersMethodprepare_tokens_with_masks
(self, x, masks=None)
segmentation/facebookresearch_dinov2_main/dinov2/models/vision_transformer.py:191
↓ 4 callersFunctionrankstr
()
segmentation/facebookresearch_dinov2_main/dinov2/fsdp/__init__.py:82
↓ 4 callersFunctionreport_progress
(params, data, i, progress_bar, iter_time_idx, sil_thres, every_i=1, qual_every_i=1, trac
utils/eval_utils.py:79
↓ 4 callersFunctionsetup
Create configs and perform basic setups.
segmentation/facebookresearch_dinov2_main/dinov2/utils/config.py:64
↓ 4 callersFunctionsubmit_jobs
(task_class, args, name: str)
segmentation/facebookresearch_dinov2_main/dinov2/run/submit.py:92
↓ 4 callersFunctiontransformed_params2rendervar
(params, w2c, transformed_pts)
utils/slam_helpers.py:81
↓ 3 callersMethod_get_entries_path
(self, root: Optional[str] = None)
segmentation/facebookresearch_dinov2_main/dinov2/data/datasets/image_net_22k.py:124
↓ 3 callersMethod_load_extra
(self, extra_path: str)
segmentation/facebookresearch_dinov2_main/dinov2/data/datasets/image_net_22k.py:186
↓ 3 callersFunction_run
(command)
segmentation/facebookresearch_dinov2_main/dinov2/utils/utils.py:50
↓ 3 callersMethod_save_extra
(self, extra_array: np.ndarray, extra_path: str)
segmentation/facebookresearch_dinov2_main/dinov2/data/datasets/image_net.py:89
↓ 3 callersFunctionall_gather_and_flatten
(tensor_rank)
segmentation/facebookresearch_dinov2_main/dinov2/eval/utils.py:87
↓ 3 callersFunctionbuild_metric
(metric_type: MetricType, *, num_classes: int, ks: Optional[tuple] = None)
segmentation/facebookresearch_dinov2_main/dinov2/eval/metrics.py:44
↓ 3 callersFunctioncalc_psnr
(img1, img2)
utils/slam_external.py:27
↓ 3 callersMethoddecode
(self)
segmentation/facebookresearch_dinov2_main/dinov2/data/datasets/decoders.py:14
↓ 3 callersFunctionevaluate
( model: nn.Module, data_loader, postprocessors: Dict[str, nn.Module], metrics: Dict[str, Metr
segmentation/facebookresearch_dinov2_main/dinov2/eval/utils.py:49
↓ 3 callersFunctionevaluate_linear_classifiers
( feature_model, linear_classifiers, data_loader, metric_type, metrics_file_path, trai
segmentation/facebookresearch_dinov2_main/dinov2/eval/linear.py:260
↓ 3 callersMethodforward_features
(self, x, masks=None)
segmentation/facebookresearch_dinov2_main/dinov2/models/vision_transformer.py:221
↓ 3 callersFunctionget_cluster_type
(cluster_type: Optional[ClusterType] = None)
segmentation/facebookresearch_dinov2_main/dinov2/utils/cluster.py:32
↓ 3 callersFunctionget_fsdp_modules
(x)
segmentation/facebookresearch_dinov2_main/dinov2/fsdp/__init__.py:73
↓ 3 callersFunctionhomogenize_points
r"""Convert a set of points to homogeneous coordinates. Args: pts (torch.Tensor): Tensor containing points to be homogenized. Shape:
datasets/gradslam_datasets/geometryutils.py:11
↓ 3 callersFunctioninitialize_optimizer
(params, lrs_dict, tracking)
sem_gauss.py:135
↓ 3 callersFunctionmake_classification_eval_transform
( *, resize_size: int = 256, interpolation=transforms.InterpolationMode.BICUBIC, crop_size: in
segmentation/facebookresearch_dinov2_main/dinov2/data/transforms.py:78
↓ 3 callersFunctionmake_normalize_transform
( mean: Sequence[float] = IMAGENET_DEFAULT_MEAN, std: Sequence[float] = IMAGENET_DEFAULT_STD, )
segmentation/facebookresearch_dinov2_main/dinov2/data/transforms.py:47
↓ 3 callersFunctionnormalize
(x)
eval_mesh/eval_recon.py:13
↓ 3 callersFunctionrelative_transformation
r"""Function that computes the relative homogenous transformation from a reference transformation :math:`T_1^{0} = \begin{bmatrix} R_1 & t_1 \\
datasets/gradslam_datasets/geometryutils.py:413
↓ 3 callersFunctionremove_ddp_wrapper
(m: nn.Module)
segmentation/facebookresearch_dinov2_main/dinov2/eval/linear.py:159
↓ 3 callersFunctionsave_params_ckpt
(output_params, output_dir, time_idx)
utils/common_utils.py:43
↓ 3 callersMethodset_mode_get_feature
(self)
utils/dinov2_seg.py:155
↓ 3 callersMethodset_mode_get_semantic
(self)
utils/dinov2_seg.py:158
↓ 3 callersFunctionsetup_and_build_model
(args)
segmentation/facebookresearch_dinov2_main/dinov2/eval/setup.py:71
↓ 3 callersMethodupdate
(self, preds: Tensor, target: Tensor)
segmentation/facebookresearch_dinov2_main/dinov2/eval/metrics.py:88
↓ 2 callersMethod__init__
(self, classifiers_dict)
segmentation/facebookresearch_dinov2_main/dinov2/eval/linear.py:205
↓ 2 callersFunction_check_env_variable
(key: str, new_value: str)
segmentation/facebookresearch_dinov2_main/dinov2/distributed/__init__.py:146
↓ 2 callersMethod_get_class_ids_path
(self, root: Optional[str] = None)
segmentation/facebookresearch_dinov2_main/dinov2/data/datasets/image_net_22k.py:127
↓ 2 callersMethod_get_extra_full_path
(self, extra_path: str)
segmentation/facebookresearch_dinov2_main/dinov2/data/datasets/image_net.py:82
↓ 2 callersMethod_get_knn_sims_and_labels
(self, similarity, train_labels)
segmentation/facebookresearch_dinov2_main/dinov2/eval/knn.py:124
↓ 2 callersFunction_get_torch_dtype
(size: int)
segmentation/facebookresearch_dinov2_main/dinov2/data/samplers.py:60
↓ 2 callersFunction_make_dinov2_model_name
(arch_name: str, patch_size: int)
segmentation/facebookresearch_dinov2_main/hubconf.py:17
↓ 2 callersMethod_save_extra
(self, extra_array: np.ndarray, extra_path: str)
segmentation/facebookresearch_dinov2_main/dinov2/data/datasets/image_net_22k.py:191
↓ 2 callersFunctionas_intrinsics_matrix
Get matrix representation of intrinsics.
datasets/gradslam_datasets/basedataset.py:34
↓ 2 callersFunctionbuild_model_from_cfg
(cfg, only_teacher=False)
segmentation/facebookresearch_dinov2_main/dinov2/models/__init__.py:40
↓ 2 callersFunctionbuild_topk_accuracy_metric
(average_type: AccuracyAveraging, num_classes: int, ks: tuple = (1, 5))
segmentation/facebookresearch_dinov2_main/dinov2/eval/metrics.py:60
↓ 2 callersFunctioncreate_linear_input
(x_tokens_list, use_n_blocks, use_avgpool)
segmentation/facebookresearch_dinov2_main/dinov2/eval/linear.py:171
↓ 2 callersFunctiondo_test
(cfg, model, iteration)
segmentation/facebookresearch_dinov2_main/dinov2/train/train.py:123
↓ 2 callersFunctiondrop_add_residual_stochastic_depth
( x: Tensor, residual_func: Callable[[Tensor], Tensor], sample_drop_ratio: float = 0.0, )
segmentation/facebookresearch_dinov2_main/dinov2/layers/block.py:110
↓ 2 callersFunctiondrop_add_residual_stochastic_depth_list
( x_list: List[Tensor], residual_func: Callable[[Tensor, Any], Tensor], sample_drop_ratio: float =
segmentation/facebookresearch_dinov2_main/dinov2/layers/block.py:181
↓ 2 callersFunctionevaluate_ate
Input : gt_traj: list of 4x4 matrices est_traj: list of 4x4 matrices len(gt_traj) == len(est_traj)
utils/eval_utils.py:60
↓ 2 callersFunctionevaluate_model
(*, logreg_model, logreg_metric, test_data_loader, device)
segmentation/facebookresearch_dinov2_main/dinov2/eval/log_regression.py:147
↓ 2 callersFunctionget_attn_bias_and_cat
this will perform the index select, cat the tensors, and provide the attn_bias from cache
segmentation/facebookresearch_dinov2_main/dinov2/layers/block.py:157
↓ 2 callersMethodget_dirname
(self, class_id: Optional[str] = None)
segmentation/facebookresearch_dinov2_main/dinov2/data/datasets/image_net.py:36
↓ 2 callersFunctionget_fsdp_wrapper
(model_cfg, modules_to_wrap=set())
segmentation/facebookresearch_dinov2_main/dinov2/fsdp/__init__.py:23
↓ 2 callersFunctionget_global_rank
Returns: The rank of the current process within the global process group.
segmentation/facebookresearch_dinov2_main/dinov2/distributed/__init__.py:36
↓ 2 callersFunctionget_global_size
Returns: The number of processes in the process group
segmentation/facebookresearch_dinov2_main/dinov2/distributed/__init__.py:28
↓ 2 callersMethodget_intermediate_layers
( self, x: torch.Tensor, n: Union[int, Sequence] = 1, # Layers or n last layers to ta
segmentation/facebookresearch_dinov2_main/dinov2/models/vision_transformer.py:264
↓ 2 callersFunctionget_pointcloud
(color, depth, sem_feature, intrinsics, w2c, transform_pts=True, mask=None, compute_mean_sq
sem_gauss.py:46
↓ 2 callersFunctionget_requirements
(path: str = HERE / "requirements.txt")
segmentation/facebookresearch_dinov2_main/setup.py:30
↓ 2 callersFunctionget_slurm_partition
(cluster_type: Optional[ClusterType] = None)
segmentation/facebookresearch_dinov2_main/dinov2/utils/cluster.py:62
↓ 2 callersFunctionload_config
(config_name: str)
segmentation/facebookresearch_dinov2_main/dinov2/configs/__init__.py:12
↓ 2 callersFunctionload_dataset_config
Loads config file. Args: path (str): path to config file. default_path (str, optional): whether to use default path. Default
datasets/gradslam_datasets/dataconfig.py:5
↓ 2 callersFunctionmake_2tuple
(x)
segmentation/facebookresearch_dinov2_main/dinov2/layers/patch_embed.py:17
↓ 2 callersFunctionmake_dinov2_model
( *, arch_name: str = "vit_large", img_size: int = 518, patch_size: int = 14, init_values:
utils/dinov2_seg.py:21
↓ 2 callersFunctionmake_eval_data_loader
(test_dataset_str, batch_size, num_workers, metric_type)
segmentation/facebookresearch_dinov2_main/dinov2/eval/linear.py:411
↓ 2 callersMethodmedian
(self)
segmentation/facebookresearch_dinov2_main/dinov2/logging/helpers.py:167
↓ 2 callersFunctionparams2cpu
(params)
utils/common_utils.py:25
↓ 2 callersMethodset_mode_classification
(self)
utils/dinov2_seg.py:161
↓ 2 callersFunctionsetup_camera
(w, h, k, w2c, near=0.01, far=100)
utils/recon_utils.py:4
↓ 2 callersMethodsinkhorn_knopp_teacher
(self, teacher_output, teacher_temp, n_iterations=3)
segmentation/facebookresearch_dinov2_main/dinov2/loss/dino_clstoken_loss.py:36
↓ 2 callersMethodsoftmax_center_teacher
(self, teacher_output, teacher_temp)
segmentation/facebookresearch_dinov2_main/dinov2/loss/dino_clstoken_loss.py:30
↓ 2 callersMethodsynchronize_between_processes
(self)
segmentation/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)
segmentation/facebookresearch_dinov2_main/dinov2/eval/log_regression.py:153
↓ 2 callersFunctiontransform_pts_3d
r"""Transforms a set of points `pts_b` from frame `b` to frame `a`, given an SE(3) transformation matrix `t_ab` Args: pts_b (torch.Te
datasets/gradslam_datasets/geometryutils.py:304
↓ 2 callersFunctionunhomogenize_points
r"""Convert a set of points from homogeneous coordinates to Euclidean coordinates. This is usually done by taking each point (x, y, z, w) and
datasets/gradslam_datasets/geometryutils.py:39
↓ 2 callersMethodupdate
(self, value, num=1)
segmentation/facebookresearch_dinov2_main/dinov2/logging/helpers.py:147
↓ 2 callersMethodupdate_center
(self, teacher_output)
segmentation/facebookresearch_dinov2_main/dinov2/loss/dino_clstoken_loss.py:78
↓ 1 callersMethod__init__
(self, model)
segmentation/facebookresearch_dinov2_main/dinov2/eval/utils.py:23
↓ 1 callersMethod__init__
(self, train_features, train_labels, nb_knn, T, device, num_classes=1000)
segmentation/facebookresearch_dinov2_main/dinov2/eval/knn.py:109
next →1–100 of 506, ranked by callers