MCPcopy Create free account

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

Functions506 in github.com/IRMVLab/SemGauss-SLAM

↓ 1 callersMethod__init__
( self, in_features: int, hidden_features: Optional[int] = None, out_features:
segmentation/facebookresearch_dinov2_main/dinov2/layers/swiglu_ffn.py:14
↓ 1 callersFunction_build_mlp
(nlayers, in_dim, bottleneck_dim, hidden_dim=None, use_bn=False, bias=True)
segmentation/facebookresearch_dinov2_main/dinov2/layers/dino_head.py:45
↓ 1 callersFunction_collect_env_vars
()
segmentation/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
segmentation/facebookresearch_dinov2_main/dinov2/logging/__init__.py:19
↓ 1 callersMethod_dump_class_ids
(self, *args, **kwargs)
segmentation/facebookresearch_dinov2_main/dinov2/data/datasets/image_net_22k.py:281
↓ 1 callersMethod_dump_class_ids_and_names
(self)
segmentation/facebookresearch_dinov2_main/dinov2/data/datasets/image_net.py:253
↓ 1 callersMethod_dump_entries
(self, *args, **kwargs)
segmentation/facebookresearch_dinov2_main/dinov2/data/datasets/image_net_22k.py:243
↓ 1 callersMethod_dump_entries
(self)
segmentation/facebookresearch_dinov2_main/dinov2/data/datasets/image_net.py:187
↓ 1 callersMethod_dump_extra
(self, *args, **kwargs)
segmentation/facebookresearch_dinov2_main/dinov2/data/datasets/image_net_22k.py:298
↓ 1 callersMethod_find_class_ids
(self, path: str)
segmentation/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)
segmentation/facebookresearch_dinov2_main/dinov2/eval/knn.py:144
↓ 1 callersFunction_generate_randperm_indices
Generate the indices of a random permutation.
segmentation/facebookresearch_dinov2_main/dinov2/data/samplers.py:64
↓ 1 callersFunction_get_available_port
()
segmentation/facebookresearch_dinov2_main/dinov2/distributed/__init__.py:102
↓ 1 callersMethod_get_class_ids
(self)
segmentation/facebookresearch_dinov2_main/dinov2/data/datasets/image_net.py:112
↓ 1 callersMethod_get_class_names
(self)
segmentation/facebookresearch_dinov2_main/dinov2/data/datasets/image_net.py:120
↓ 1 callersMethod_get_intermediate_layers_chunked
(self, x, n=1)
segmentation/facebookresearch_dinov2_main/dinov2/models/vision_transformer.py:250
↓ 1 callersMethod_get_intermediate_layers_not_chunked
(self, x, n=1)
segmentation/facebookresearch_dinov2_main/dinov2/models/vision_transformer.py:238
↓ 1 callersFunction_get_master_port
(seed: int = 0)
segmentation/facebookresearch_dinov2_main/dinov2/distributed/__init__.py:91
↓ 1 callersFunction_get_numpy_dtype
(size: int)
segmentation/facebookresearch_dinov2_main/dinov2/data/samplers.py:56
↓ 1 callersFunction_get_tarball_path
(class_id: str)
segmentation/facebookresearch_dinov2_main/dinov2/data/datasets/image_net_22k.py:56
↓ 1 callersFunction_guess_cluster_type
()
segmentation/facebookresearch_dinov2_main/dinov2/utils/cluster.py:19
↓ 1 callersFunction_is_slurm_job_process
()
segmentation/facebookresearch_dinov2_main/dinov2/distributed/__init__.py:125
↓ 1 callersMethod_iterator
(self)
segmentation/facebookresearch_dinov2_main/dinov2/data/samplers.py:105
↓ 1 callersMethod_iterator
(self)
segmentation/facebookresearch_dinov2_main/dinov2/data/samplers.py:202
↓ 1 callersMethod_load_entries_class_ids
(self, root: Optional[str] = None)
segmentation/facebookresearch_dinov2_main/dinov2/data/datasets/image_net_22k.py:142
↓ 1 callersMethod_load_labels
(self, labels_path: str)
segmentation/facebookresearch_dinov2_main/dinov2/data/datasets/image_net.py:172
↓ 1 callersFunction_make_dinov2_linear_head
( *, model_name: str = "dinov2_vitl14", embed_dim: int = 1024, layers: int = 4, pretrained
segmentation/facebookresearch_dinov2_main/hubconf.py:82
↓ 1 callersFunction_make_mmap_tarball
(tarballs_root: str, mmap_cache_size: int)
segmentation/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,
segmentation/facebookresearch_dinov2_main/dinov2/data/loaders.py:101
↓ 1 callersFunction_make_seed
(seed: int, start: int, iter_count: int)
segmentation/facebookresearch_dinov2_main/dinov2/data/samplers.py:161
↓ 1 callersMethod_mask
(self, mask, max_mask_patches)
segmentation/facebookresearch_dinov2_main/dinov2/data/masking.py:50
↓ 1 callersFunction_parse_dataset_str
(dataset_str: str)
segmentation/facebookresearch_dinov2_main/dinov2/data/loaders.py:45
↓ 1 callersFunction_parse_slurm_node_list
(s: str)
segmentation/facebookresearch_dinov2_main/dinov2/distributed/__init__.py:129
↓ 1 callersMethod_preprocess_color
r"""Preprocesses the color image by resizing to :math:`(H, W, C)`, (optionally) normalizing values to :math:`[0, 1]`, and (optionally) using c
datasets/gradslam_datasets/basedataset.py:202
↓ 1 callersMethod_preprocess_depth
r"""Preprocesses the depth image by resizing, adding channel dimension, and scaling values to meters. Optionally converts depth from channels
datasets/gradslam_datasets/basedataset.py:235
↓ 1 callersMethod_preprocess_poses
r"""Preprocesses the poses by setting first pose in a sequence to identity and computing the relative homogenous transformation for all other
datasets/gradslam_datasets/basedataset.py:259
↓ 1 callersMethod_preprocess_semantic
(self, semantic: np.ndarray)
datasets/gradslam_datasets/basedataset.py:227
↓ 1 callersFunction_restrict_print_to_main_process
This function disables printing when not in the main process
segmentation/facebookresearch_dinov2_main/dinov2/distributed/__init__.py:75
↓ 1 callersMethod_set_from_local
(self)
segmentation/facebookresearch_dinov2_main/dinov2/distributed/__init__.py:211
↓ 1 callersMethod_set_from_preset_env
(self)
segmentation/facebookresearch_dinov2_main/dinov2/distributed/__init__.py:199
↓ 1 callersMethod_set_from_slurm_env
(self)
segmentation/facebookresearch_dinov2_main/dinov2/distributed/__init__.py:182
↓ 1 callersMethod_setup_args
(self)
segmentation/facebookresearch_dinov2_main/dinov2/run/eval/linear.py:36
↓ 1 callersMethod_setup_args
(self)
segmentation/facebookresearch_dinov2_main/dinov2/run/eval/log_regression.py:36
↓ 1 callersMethod_setup_args
(self)
segmentation/facebookresearch_dinov2_main/dinov2/run/eval/knn.py:36
↓ 1 callersMethod_setup_args
(self)
segmentation/facebookresearch_dinov2_main/dinov2/run/train/train.py:36
↓ 1 callersMethod_shuffled_iterator
(self)
segmentation/facebookresearch_dinov2_main/dinov2/data/samplers.py:112
↓ 1 callersMethod_shuffled_iterator
(self)
segmentation/facebookresearch_dinov2_main/dinov2/data/samplers.py:209
↓ 1 callersMethod_similarity_for_rank
(self, features_rank, source_rank)
segmentation/facebookresearch_dinov2_main/dinov2/eval/knn.py:129
↓ 1 callersFunction_sqrt_positive_part
Returns torch.sqrt(torch.max(0, x)) but with a zero subgradient where x is 0. Source: https://pytorch3d.readthedocs.io/en/latest/_modules
utils/slam_helpers.py:8
↓ 1 callersFunction_ssim
(img1, img2, window, window_size, channel, size_average=True)
utils/slam_external.py:53
↓ 1 callersFunctionadd_new_gaussians
(params, variables, curr_data, sil_thres, time_idx, mean_sq_dist_method, sem_feature)
sem_gauss.py:363
↓ 1 callersFunctionadd_residual
(x, brange, residual, residual_scale_factor, scaling_vector=None)
segmentation/facebookresearch_dinov2_main/dinov2/layers/block.py:142
↓ 1 callersFunctionalign
Align two trajectories using the method of Horn (closed-form). Args: model -- first trajectory (3xn) data -- second trajectory (3
utils/eval_utils.py:23
↓ 1 callersMethodapply_center_update
(self)
segmentation/facebookresearch_dinov2_main/dinov2/loss/ibot_patch_loss.py:142
↓ 1 callersMethodapply_center_update
(self)
segmentation/facebookresearch_dinov2_main/dinov2/loss/dino_clstoken_loss.py:90
↓ 1 callersFunctionapply_optim_scheduler
(optimizer, lr, wd, last_layer_lr)
segmentation/facebookresearch_dinov2_main/dinov2/train/train.py:114
↓ 1 callersFunctionapply_scaling_rules_to_cfg
(cfg)
segmentation/facebookresearch_dinov2_main/dinov2/utils/config.py:22
↓ 1 callersFunctionas_torch_dtype
(dtype: TypeSpec)
segmentation/facebookresearch_dinov2_main/dinov2/utils/dtype.py:32
↓ 1 callersMethodbackprop_loss
(self, loss)
segmentation/facebookresearch_dinov2_main/dinov2/train/ssl_meta_arch.py:129
↓ 1 callersFunctionbatch_intersection_union
mIoU
utils/segmentationMetric.py:88
↓ 1 callersFunctionbatch_pix_accuracy
PixAcc
utils/segmentationMetric.py:74
↓ 1 callersFunctionbuild_model
(args, only_teacher=False, img_size=224)
segmentation/facebookresearch_dinov2_main/dinov2/models/__init__.py:15
↓ 1 callersFunctionbuild_model_for_eval
(config, pretrained_weights)
segmentation/facebookresearch_dinov2_main/dinov2/eval/setup.py:63
↓ 1 callersFunctionbuild_optimizer
(cfg, params_groups)
segmentation/facebookresearch_dinov2_main/dinov2/train/train.py:62
↓ 1 callersFunctionbuild_schedulers
(cfg)
segmentation/facebookresearch_dinov2_main/dinov2/train/train.py:66
↓ 1 callersFunctionbuild_topk_imagenet_real_accuracy_metric
(num_classes: int, ks: tuple = (1, 5))
segmentation/facebookresearch_dinov2_main/dinov2/eval/metrics.py:67
↓ 1 callersFunctioncalc_ssim
(img1, img2, window_size=11, size_average=True)
utils/slam_external.py:43
↓ 1 callersFunctioncam2pixel
r"""Transforms coordinates from the camera frame to the pixel frame. # based on # https://github.com/ClementPinard/SfmLearner-Pytorch/blob/ma
datasets/gradslam_datasets/geometryutils.py:611
↓ 1 callersFunctioncheck_proj
Check if points can be projected into the camera view. Returns: bool: True if there are points can be projected
eval_mesh/eval_recon.py:34
↓ 1 callersFunctioncompose_transforms_3d
r"""Compose two homogeneous 3D transforms. Args: trans1 (torch.Tensor): first transformation (shape: :math:`* \times 4 \times
datasets/gradslam_datasets/geometryutils.py:244
↓ 1 callersMethodcompute
(self)
segmentation/facebookresearch_dinov2_main/dinov2/eval/metrics.py:112
↓ 1 callersMethodcompute_neighbors
(self, features_rank)
segmentation/facebookresearch_dinov2_main/dinov2/eval/knn.py:162
↓ 1 callersFunctioncreate_class_indices_mapping
(labels)
segmentation/facebookresearch_dinov2_main/dinov2/eval/knn.py:234
↓ 1 callersFunctioncreate_meshgrid
r"""Generates a coordinate grid for an image. When `normalized_coords` is set to True, the grid is normalized to be in the range [-1, 1] (to
datasets/gradslam_datasets/geometryutils.py:576
↓ 1 callersFunctioncreate_module_dict
(*, module, n_per_class_list, n_tries, nb_knn, train_features, train_labels)
segmentation/facebookresearch_dinov2_main/dinov2/eval/knn.py:198
↓ 1 callersFunctioncreate_window
(window_size, channel)
utils/slam_external.py:37
↓ 1 callersFunctiondefault_setup
(args)
segmentation/facebookresearch_dinov2_main/dinov2/utils/config.py:50
↓ 1 callersFunctiondense_semantic_slam
(config: dict)
sem_gauss.py:430
↓ 1 callersFunctiondo_train
(cfg, model, resume=False)
segmentation/facebookresearch_dinov2_main/dinov2/train/train.py:135
↓ 1 callersFunctiondrop_path
(x, drop_prob: float = 0.0, training: bool = False)
segmentation/facebookresearch_dinov2_main/dinov2/layers/drop_path.py:15
↓ 1 callersMethoddump_in_output_file
(self, iteration, iter_time, data_time)
segmentation/facebookresearch_dinov2_main/dinov2/logging/helpers.py:54
↓ 1 callersFunctioneval
(config, dataset, final_params, num_frames, eval_dir, sil_thres, mapping_iters, add_new_gaussians, ev
utils/eval_utils.py:172
↓ 1 callersFunctioneval_knn
( model, train_dataset, val_dataset, accuracy_averaging, nb_knn, temperature, batc
segmentation/facebookresearch_dinov2_main/dinov2/eval/knn.py:245
↓ 1 callersFunctioneval_knn_with_model
( model, output_dir, train_dataset_str="ImageNet:split=TRAIN", val_dataset_str="ImageNet:split
segmentation/facebookresearch_dinov2_main/dinov2/eval/knn.py:318
↓ 1 callersFunctioneval_linear
( *, feature_model, linear_classifiers, train_data_loader, val_data_loader, metrics_fi
segmentation/facebookresearch_dinov2_main/dinov2/eval/linear.py:313
↓ 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
segmentation/facebookresearch_dinov2_main/dinov2/eval/log_regression.py:252
↓ 1 callersFunctioneval_log_regression_with_model
( model, train_dataset_str="ImageNet:split=TRAIN", val_dataset_str="ImageNet:split=VAL", finet
segmentation/facebookresearch_dinov2_main/dinov2/eval/log_regression.py:362
↓ 1 callersMethodexport
(self, *, overwrite: bool)
segmentation/facebookresearch_dinov2_main/dinov2/distributed/__init__.py:220
↓ 1 callersFunctionextract_features_with_dataloader
(model, data_loader, sample_count, gather_on_cpu=False)
segmentation/facebookresearch_dinov2_main/dinov2/eval/utils.py:114
↓ 1 callersFunctionfilter_train
(mapping, n_per_class, seed)
segmentation/facebookresearch_dinov2_main/dinov2/eval/knn.py:225
↓ 1 callersMethodfit
(self, train_features, train_labels)
segmentation/facebookresearch_dinov2_main/dinov2/eval/log_regression.py:137
↓ 1 callersMethodforward
(self, x: Tensor)
segmentation/facebookresearch_dinov2_main/dinov2/layers/block.py:82
↓ 1 callersMethodforward
(self, x: Tensor)
segmentation/facebookresearch_dinov2_main/dinov2/layers/attention.py:49
↓ 1 callersMethodforward_backward
(self, images, teacher_temp)
segmentation/facebookresearch_dinov2_main/dinov2/train/ssl_meta_arch.py:135
↓ 1 callersMethodforward_features_list
(self, x_list, masks_list)
segmentation/facebookresearch_dinov2_main/dinov2/models/vision_transformer.py:202
↓ 1 callersMethodforward_masked
( self, student_patch_tokens_masked, teacher_patch_tokens_masked, student_mask
segmentation/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
segmentation/facebookresearch_dinov2_main/dinov2/layers/block.py:205
↓ 1 callersFunctionfree_if_fsdp
(x)
segmentation/facebookresearch_dinov2_main/dinov2/fsdp/__init__.py:66
↓ 1 callersMethodfsdp_synchronize_streams
(self)
segmentation/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"))
segmentation/facebookresearch_dinov2_main/dinov2/utils/param_groups.py:83
← previousnext →101–200 of 506, ranked by callers