MCPcopy Create free account

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

Functions506 in github.com/IRMVLab/SemGauss-SLAM

↓ 1 callersFunctiongaussian
(window_size, sigma)
utils/slam_external.py:32
↓ 1 callersFunctionget_align_transformation
Get the transformation matrix to align the reconstructed mesh to the ground truth mesh.
eval_mesh/eval_recon.py:17
↓ 1 callersFunctionget_args_parser
( description: Optional[str] = None, parents: Optional[List[argparse.ArgumentParser]] = [], add_he
segmentation/facebookresearch_dinov2_main/dinov2/eval/linear.py:34
↓ 1 callersFunctionget_args_parser
( description: Optional[str] = None, parents: Optional[List[argparse.ArgumentParser]] = [], add_he
segmentation/facebookresearch_dinov2_main/dinov2/eval/log_regression.py:39
↓ 1 callersFunctionget_args_parser
( description: Optional[str] = None, parents: Optional[List[argparse.ArgumentParser]] = [], add_he
segmentation/facebookresearch_dinov2_main/dinov2/eval/knn.py:30
↓ 1 callersFunctionget_args_parser
(add_help: bool = True)
segmentation/facebookresearch_dinov2_main/dinov2/train/train.py:31
↓ 1 callersFunctionget_autocast_dtype
(config)
segmentation/facebookresearch_dinov2_main/dinov2/eval/setup.py:53
↓ 1 callersFunctionget_branges_scales
(x, sample_drop_ratio=0.0)
segmentation/facebookresearch_dinov2_main/dinov2/layers/block.py:134
↓ 1 callersFunctionget_cam_position
(gt_meshfile)
eval_mesh/eval_recon.py:66
↓ 1 callersFunctionget_cfg_from_args
(args)
segmentation/facebookresearch_dinov2_main/dinov2/utils/config.py:41
↓ 1 callersFunctionget_checkpoint_path
(cluster_type: Optional[ClusterType] = None)
segmentation/facebookresearch_dinov2_main/dinov2/utils/cluster.py:39
↓ 1 callersMethodget_class_id
(self, index: int)
segmentation/facebookresearch_dinov2_main/dinov2/data/datasets/image_net.py:157
↓ 1 callersFunctionget_dataset
(config_dict, basedir, sequence, **kwargs)
sem_gauss.py:37
↓ 1 callersFunctionget_depth_and_silhouette
Function to compute depth and silhouette for each gaussian. These are evaluated at gaussian center.
utils/slam_helpers.py:95
↓ 1 callersMethodget_dinov2
(self)
utils/dinov2_seg.py:149
↓ 1 callersMethodget_filepaths
Return paths to color images, depth images. Implement in subclass.
datasets/gradslam_datasets/basedataset.py:194
↓ 1 callersMethodget_image_data
(self, index: int)
segmentation/facebookresearch_dinov2_main/dinov2/data/datasets/extended.py:18
↓ 1 callersMethodget_image_relpath
(self, actual_index: int, class_id: Optional[str] = None)
segmentation/facebookresearch_dinov2_main/dinov2/data/datasets/image_net.py:39
↓ 1 callersMethodget_maybe_fused_params_for_submodel
(self, m)
segmentation/facebookresearch_dinov2_main/dinov2/train/ssl_meta_arch.py:374
↓ 1 callersFunctionget_package_version
()
segmentation/facebookresearch_dinov2_main/setup.py:43
↓ 1 callersMethodget_params_groups
(self)
segmentation/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)
segmentation/facebookresearch_dinov2_main/dinov2/utils/param_groups.py:42
↓ 1 callersFunctionget_pointcloud
(depth, intrinsics, w2c, sampled_indices)
utils/keyframe_selection.py:10
↓ 1 callersMethodget_shape
(self)
segmentation/facebookresearch_dinov2_main/dinov2/data/masking.py:47
↓ 1 callersFunctionget_shared_folder
()
segmentation/facebookresearch_dinov2_main/dinov2/run/submit.py:83
↓ 1 callersFunctionget_slurm_executor_parameters
( nodes: int, num_gpus_per_node: int, cluster_type: Optional[ClusterType] = None, **kwargs )
segmentation/facebookresearch_dinov2_main/dinov2/utils/cluster.py:75
↓ 1 callersMethodget_target
(self, index: int)
segmentation/facebookresearch_dinov2_main/dinov2/data/datasets/extended.py:21
↓ 1 callersMethodget_targets
(self)
segmentation/facebookresearch_dinov2_main/dinov2/data/datasets/image_net.py:153
↓ 1 callersFunctionget_user_checkpoint_path
(cluster_type: Optional[ClusterType] = None)
segmentation/facebookresearch_dinov2_main/dinov2/utils/cluster.py:52
↓ 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
segmentation/facebookresearch_dinov2_main/dinov2/utils/param_groups.py:14
↓ 1 callersFunctionhas_batchnorms
(model)
segmentation/facebookresearch_dinov2_main/dinov2/utils/utils.py:91
↓ 1 callersFunctionhas_ddp_wrapper
(m: nn.Module)
segmentation/facebookresearch_dinov2_main/dinov2/eval/linear.py:155
↓ 1 callersMethodinit_weights
(self)
segmentation/facebookresearch_dinov2_main/dinov2/models/vision_transformer.py:160
↓ 1 callersFunctioninitialize_camera_pose
(params, curr_time_idx, forward_prop)
sem_gauss.py:409
↓ 1 callersFunctioninitialize_first_timestep
(dataset, num_frames, scene_radius_depth_ratio, mean_sq_dist_method, seg_net)
sem_gauss.py:143
↓ 1 callersFunctioninitialize_new_params
(new_pt_cld, mean3_sq_dist)
sem_gauss.py:341
↓ 1 callersFunctioninitialize_params
(init_pt_cld, num_frames, mean3_sq_dist)
sem_gauss.py:100
↓ 1 callersMethodinterpolate_pos_encoding
(self, x, w, h)
segmentation/facebookresearch_dinov2_main/dinov2/models/vision_transformer.py:165
↓ 1 callersFunctioninverse_sigmoid
(x)
utils/slam_external.py:113
↓ 1 callersFunctioninverse_transfom_3d
r"""Inverts a 4 x 4 3D transformation matrix. Args: trans (torch.Tensor): transformation matrix (shape: :math:`* \times 4 \ti
datasets/gradslam_datasets/geometryutils.py:205
↓ 1 callersFunctionis_fsdp
(x)
segmentation/facebookresearch_dinov2_main/dinov2/fsdp/__init__.py:58
↓ 1 callersFunctionis_main_process
Returns: True if the current process is the main one.
segmentation/facebookresearch_dinov2_main/dinov2/distributed/__init__.py:67
↓ 1 callersFunctionis_sharded_fsdp
(x)
segmentation/facebookresearch_dinov2_main/dinov2/fsdp/__init__.py:62
↓ 1 callersFunctionkeyframe_selection_overlap
Select overlapping keyframes to the current camera observation. Args: gt_depth (tensor): ground truth depth image of the
utils/keyframe_selection.py:40
↓ 1 callersFunctionl1_loss_v1
(x, y)
utils/slam_helpers.py:5
↓ 1 callersMethodload_poses
Load camera poses. Implement in subclass.
datasets/gradslam_datasets/basedataset.py:198
↓ 1 callersFunctionlossfunc
(t, s, temp)
segmentation/facebookresearch_dinov2_main/dinov2/loss/ibot_patch_loss.py:21
↓ 1 callersFunctionmain
(args)
segmentation/facebookresearch_dinov2_main/dinov2/eval/linear.py:595
↓ 1 callersFunctionmain
(args)
segmentation/facebookresearch_dinov2_main/dinov2/eval/log_regression.py:423
↓ 1 callersFunctionmain
(args)
segmentation/facebookresearch_dinov2_main/dinov2/eval/knn.py:379
↓ 1 callersFunctionmain
(args)
segmentation/facebookresearch_dinov2_main/dinov2/train/train.py:298
↓ 1 callersFunctionmain
()
segmentation/facebookresearch_dinov2_main/dinov2/run/eval/linear.py:45
↓ 1 callersFunctionmain
()
segmentation/facebookresearch_dinov2_main/dinov2/run/eval/log_regression.py:45
↓ 1 callersFunctionmain
()
segmentation/facebookresearch_dinov2_main/dinov2/run/eval/knn.py:45
↓ 1 callersFunctionmain
()
segmentation/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
segmentation/facebookresearch_dinov2_main/dinov2/data/transforms.py:56
↓ 1 callersFunctionmatrix_to_quaternion
Convert rotations given as rotation matrices to quaternions. Args: matrix: Rotation matrices as tensor of shape (..., 3, 3). Re
utils/slam_helpers.py:19
↓ 1 callersFunctionnamed_apply
(fn: Callable, module: nn.Module, name="", depth_first=True, include_root=False)
segmentation/facebookresearch_dinov2_main/dinov2/models/vision_transformer.py:27
↓ 1 callersFunctionnormalize_quaternion
r"""Normalize a quaternion. The quaternion should be in (x, y, z, w) format. Args: quaternion (torch.Tensor): Quaternion to be normal
datasets/gradslam_datasets/geometryutils.py:127
↓ 1 callersMethodpairwise_NNs_inner
Pairwise nearest neighbors for L2-normalized vectors. Uses Torch rather than Faiss to remain on GPU.
segmentation/facebookresearch_dinov2_main/dinov2/loss/koleo_loss.py:26
↓ 1 callersMethodparse_image_relpath
(self, image_relpath: str)
segmentation/facebookresearch_dinov2_main/dinov2/data/datasets/image_net.py:47
↓ 1 callersFunctionpixel2cam
r"""Transforms points from the pixel frame to the camera frame. Args: depth (torch.Tensor): the source depth maps (shape: :ma
datasets/gradslam_datasets/geometryutils.py:655
↓ 1 callersMethodprepare_for_distributed_training
(self)
segmentation/facebookresearch_dinov2_main/dinov2/train/ssl_meta_arch.py:393
↓ 1 callersFunctionprune_gaussians
(params, variables, optimizer, iter, prune_dict)
utils/slam_external.py:117
↓ 1 callersFunctionrasterize_gaussians
( means3D, means2D, sh, colors_precomp, opacities, scales, rotations, cov3Ds_p
diff-gaussian-rasterization-w-depth_sem_gauss/diff_gaussian_rasterization/__init__.py:17
↓ 1 callersFunctionreadEXR_onlydepth
Read depth data from EXR image file. Args: filename (str): File path. Returns: Y (numpy.array): Depth buffer in float32
datasets/gradslam_datasets/basedataset.py:46
↓ 1 callersMethodread_embedding_from_file
Read embedding from file and process it. To be implemented in subclass for each dataset separately.
datasets/gradslam_datasets/basedataset.py:290
↓ 1 callersMethodreduce_center_update
(self, teacher_patch_tokens)
segmentation/facebookresearch_dinov2_main/dinov2/loss/ibot_patch_loss.py:134
↓ 1 callersMethodreduce_center_update
(self, teacher_output)
segmentation/facebookresearch_dinov2_main/dinov2/loss/dino_clstoken_loss.py:82
↓ 1 callersFunctionremove_points
(to_remove, params, variables, optimizer)
utils/slam_external.py:89
↓ 1 callersMethodreset
Resets the internal evaluation result to initial state.
utils/segmentationMetric.py:65
↓ 1 callersFunctionreshard_fsdp_model
(x)
segmentation/facebookresearch_dinov2_main/dinov2/fsdp/__init__.py:77
↓ 1 callersFunctionrun_eval_linear
( model, output_dir, train_dataset_str, val_dataset_str, batch_size, epochs, epoch
segmentation/facebookresearch_dinov2_main/dinov2/eval/linear.py:463
↓ 1 callersFunctionsave_params
(output_params, output_dir)
utils/common_utils.py:34
↓ 1 callersFunctionscale_lr
(learning_rates, batch_size)
segmentation/facebookresearch_dinov2_main/dinov2/eval/linear.py:231
↓ 1 callersFunctionseed_everything
Set the `seed` value for torch and numpy seeds. Also turns on deterministic execution for cudnn. Parameters:
utils/common_utils.py:9
↓ 1 callersFunctionsetup_linear_classifiers
(sample_output, n_last_blocks_list, learning_rates, batch_size, num_classes=1000)
segmentation/facebookresearch_dinov2_main/dinov2/eval/linear.py:235
↓ 1 callersFunctionsweep_C_values
( *, train_features, train_labels, test_data_loader, metric_type, num_classes, tra
segmentation/facebookresearch_dinov2_main/dinov2/eval/log_regression.py:187
↓ 1 callersMethodtag_last_checkpoint
Tag the last checkpoint. Args: last_filename_basename (str): the basename of the last filename.
segmentation/facebookresearch_dinov2_main/dinov2/fsdp/__init__.py:144
↓ 1 callersFunctiontest_on_datasets
( feature_model, linear_classifiers, test_dataset_strs, batch_size, num_workers, test_
segmentation/facebookresearch_dinov2_main/dinov2/eval/linear.py:429
↓ 1 callersMethodtrain
(self)
segmentation/facebookresearch_dinov2_main/dinov2/train/ssl_meta_arch.py:370
↓ 1 callersFunctiontrain_and_evaluate
( *, C, max_iter, train_features, train_labels, logreg_metric, test_data_loader,
segmentation/facebookresearch_dinov2_main/dinov2/eval/log_regression.py:159
↓ 1 callersFunctiontransformPoint4x3
diff-gaussian-rasterization-w-depth_sem_gauss/cuda_rasterizer/auxiliary.h:58
↓ 1 callersFunctiontransformPoint4x4
diff-gaussian-rasterization-w-depth_sem_gauss/cuda_rasterizer/auxiliary.h:68
↓ 1 callersFunctiontransform_pts_nd_KF
r"""Applies a transform to a set of points. Args: pts (torch.Tensor): Points to be transformed (shape: B x N x D) (N points,
datasets/gradslam_datasets/geometryutils.py:361
↓ 1 callersFunctionupdate_params_and_optimizer
(new_params, params, optimizer)
utils/slam_external.py:75
↓ 1 callersFunctionupdate_recursive
Update two config dictionaries recursively. Args: dict1 (dict): first dictionary to be updated. dict2 (dict): second diction
datasets/gradslam_datasets/dataconfig.py:40
↓ 1 callersMethodupdate_teacher
(self, m)
segmentation/facebookresearch_dinov2_main/dinov2/train/ssl_meta_arch.py:359
↓ 1 callersFunctionviewmatrix
(z, up, pos)
eval_mesh/eval_recon.py:77
↓ 1 callersFunctionwrite_config
(cfg, output_dir, name="config.yaml")
segmentation/facebookresearch_dinov2_main/dinov2/utils/config.py:33
FunctionPYBIND11_MODULE
diff-gaussian-rasterization-w-depth_sem_gauss/ext.cpp:15
Method__call__
(self, num_masking_patches=0)
segmentation/facebookresearch_dinov2_main/dinov2/data/masking.py:74
Method__call__
Args: pic (PIL Image, numpy.ndarray or torch.tensor): Image to be converted to tensor. Returns: Tensor: Conve
segmentation/facebookresearch_dinov2_main/dinov2/data/transforms.py:30
Method__call__
(self, image)
segmentation/facebookresearch_dinov2_main/dinov2/data/augmentations.py:97
Method__call__
(self)
segmentation/facebookresearch_dinov2_main/dinov2/run/eval/linear.py:23
Method__call__
(self)
segmentation/facebookresearch_dinov2_main/dinov2/run/eval/log_regression.py:23
Method__call__
(self)
segmentation/facebookresearch_dinov2_main/dinov2/run/eval/knn.py:23
Method__call__
(self)
segmentation/facebookresearch_dinov2_main/dinov2/run/train/train.py:23
Method__getattr__
(self, attr)
segmentation/facebookresearch_dinov2_main/dinov2/logging/helpers.py:34
Method__getitem__
(self, index)
datasets/gradslam_datasets/basedataset.py:296
← previousnext →201–300 of 506, ranked by callers