MCPcopy Create free account

hub / github.com/MaverickPeter/vDiSCO / functions

Functions910 in github.com/MaverickPeter/vDiSCO

↓ 1 callersFunctionfftshift2d
(x)
prnet/models/localizer/fusion.py:40
↓ 1 callersFunctionfftshift2d
(x)
prnet/models/localizer/deformable.py:39
↓ 1 callersMethodfilter
(self, ts: np.ndarray, poses: np.ndarray)
prnet/datasets/oxford/oxford_raw.py:284
↓ 1 callersMethodfilter
(self, ts: np.ndarray, poses: np.ndarray)
prnet/datasets/nclt/nclt_raw.py:191
↓ 1 callersMethodfilter_pretrained_dict
Removes layers from pretrained state dict that are not used or changed in model Args: model_dict: dict, Default model sta
prnet/models/backbones_3d/vfe/image_vfe_modules/ffn/ddn/ddn_template.py:85
↓ 1 callersFunctionfind_nearest_ndx
(ts, timestamps)
prnet/datasets/nclt/utils.py:43
↓ 1 callersMethodflow2color
:param flow: Optical flow tensor. :return: RGB image normalized between 0 and 1.
prnet/utils/vox_utils/improc.py:235
↓ 1 callersMethodforward
:param xyz: (N1 + N2 ..., 3) tensor of the xyz coordinates of the features :param xyz_batch_cnt: (batch_size), [N1, N2, ...]
prnet/ops/pointnet2/pointnet2_stack/pointnet2_modules.py:78
↓ 1 callersMethodforward_fft
(self, input)
prnet/models/localizer/deformable_fusion.py:370
↓ 1 callersMethodforward_fft
(self, input)
prnet/models/localizer/fusion.py:361
↓ 1 callersMethodforward_fft
(self, input)
prnet/models/localizer/deformable.py:294
↓ 1 callersMethodforward_post
(self, src, pos: Optional[Tensor] = None)
prnet/models/model_utils/mppnet_utils.py:302
↓ 1 callersMethodforward_pre
(self, src, pos: Optional[Tensor] = None)
prnet/models/model_utils/mppnet_utils.py:347
↓ 1 callersFunctiongaussian2D
(shape, sigma=1)
prnet/models/model_utils/centernet_utils.py:38
↓ 1 callersMethodgem
(self, x, p=3, eps=1e-6)
prnet/models/localizer/deformable_fusion.py:577
↓ 1 callersMethodgem
(self, x, p=3, eps=1e-6)
prnet/models/localizer/fusion.py:567
↓ 1 callersMethodgem
(self, x, p=3, eps=1e-6)
prnet/models/localizer/netvlad.py:324
↓ 1 callersMethodgem
(self, x, p=3, eps=1e-6)
prnet/models/aggregation/GeM.py:15
↓ 1 callersMethodgenerate_batches
(self)
prnet/datasets/samplers.py:90
↓ 1 callersFunctiongenerate_evaluation_set
(dataset_root: str, map_sequence: str, query_sequence: str, sampling_distance: float = 0.2,
prnet/datasets/oxford/generate_evaluation_sets.py:31
↓ 1 callersFunctiongenerate_evaluation_set
(dataset_root: str, map_sequence: str, query_sequence: str, sampling_distance: float = 0.2,
prnet/datasets/nclt/generate_evaluation_sets.py:26
↓ 1 callersFunctiongenerate_image_meta_pickle
(dataset_root: str)
prnet/datasets/oxford/generate_training_tuples.py:28
↓ 1 callersFunctiongenerate_image_meta_pickle
(dataset_root: str)
prnet/datasets/nclt/generate_training_tuples.py:145
↓ 1 callersFunctionget_calib_from_file
(calib_file)
prnet/utils/calibration_kitti.py:4
↓ 1 callersFunctionget_clist_from_lrtlist
(lrtlist)
prnet/utils/vox_utils/geom.py:214
↓ 1 callersFunctionget_datetime
()
prnet/utils/params.py:118
↓ 1 callersMethodget_dense_voxels_by_center
Args: point_centers: (N, 3) max_neighbour_distance: float num_voxels: [num_x, num_y, num_z] Retu
prnet/ops/pointnet2/pointnet2_stack/pointnet2_modules.py:337
↓ 1 callersFunctionget_dist_info
(return_gpu_per_machine=False)
prnet/utils/common_utils.py:189
↓ 1 callersFunctionget_extensions
()
prnet/ops/setup.py:23
↓ 1 callersMethodget_feat_occupancy
(self, xyz, feat, Z, Y, X, clean_eps=0, xyz_zero=None)
prnet/utils/vox_utils/vox.py:219
↓ 1 callersMethodget_features
(self, x)
prnet/models/localizer/deformable_fusion.py:779
↓ 1 callersMethodget_features
(self, x)
prnet/models/localizer/fusion.py:769
↓ 1 callersMethodget_features
(self, x)
prnet/models/localizer/deformable.py:175
↓ 1 callersMethodget_features
(self, x)
prnet/models/localizer/netvlad.py:526
↓ 1 callersFunctionget_git_commit_number
()
prnet/__init__.py:11
↓ 1 callersMethodget_map_poses
(self)
prnet/datasets/base_datasets.py:186
↓ 1 callersFunctionget_max_per_row
(mat, mask)
prnet/models/loss.py:116
↓ 1 callersFunctionget_min_per_row
(mat, mask)
prnet/models/loss.py:123
↓ 1 callersMethodget_model
Get model Args: constructor: function, Model constructor Returns: model: nn.Module, Model
prnet/models/backbones_3d/focal_sparse_conv/SemanticSeg/sem_deeplabv3.py:41
↓ 1 callersMethodget_model
Get model Args: constructor: function, Model constructor Returns: model: nn.Module, Model
prnet/models/backbones_3d/vfe/image_vfe_modules/ffn/ddn/ddn_template.py:47
↓ 1 callersFunctionget_model_name
(model)
prnet/datasets/oxford/generate_training_tuples.py:21
↓ 1 callersMethodget_occupancy
(self, xyz, Z, Y, X, clean_eps=0, xyz_zero=None)
prnet/utils/vox_utils/vox.py:157
↓ 1 callersMethodget_paddings_indicator
(self, actual_num, max_num, axis=0)
prnet/models/backbones_3d/vfe/pillar_vfe.py:86
↓ 1 callersMethodget_params
(self, coords)
prnet/datasets/augmentation.py:262
↓ 1 callersMethodget_query_poses
(self)
prnet/datasets/base_datasets.py:193
↓ 1 callersFunctionget_rank
()
prnet/utils/commu_utils.py:23
↓ 1 callersMethodget_ref_T_mem
(self, B, Z, Y, X, assert_cube=False, device='cuda')
prnet/utils/vox_utils/vox.py:107
↓ 1 callersMethodget_sampled_points
Args: batch_dict: Returns: keypoints: (N1 + N2 + ..., 4), where 4 indicates [bs_idx, x, y, z]
prnet/models/backbones_3d/pfe/voxel_set_abstraction.py:227
↓ 1 callersFunctionget_xyzlist_from_lenlist
(lenlist)
prnet/utils/vox_utils/geom.py:179
↓ 1 callersMethodgrid_to_lidar_unproject
Calculate grid to LiDAR unprojection for each plane Args: pc_min: [x_min, y_min, z_min], Minimum of point cloud range (m)
prnet/models/backbones_3d/vfe/image_vfe_modules/f2v/frustum_grid_generator.py:60
↓ 1 callersFunctionicp
(anchor_pc, positive_pc, transform: np.ndarray = None, point2plane: bool = False, inlier_dist_threshol
prnet/utils/data_utils/point_clouds.py:30
↓ 1 callersFunctionin_test_split
(pos)
prnet/datasets/oxford/utils.py:34
↓ 1 callersFunctionin_test_split
(pos)
prnet/datasets/nclt/utils.py:34
↓ 1 callersFunctionin_train_split
(pos)
prnet/datasets/oxford/utils.py:25
↓ 1 callersFunctionin_train_split
(pos)
prnet/datasets/nclt/utils.py:25
↓ 1 callersMethodinit_weights
Default initialization for Parameters of Module.
prnet/ops/modules/ms_deform_attn.py:190
↓ 1 callersMethodinit_weights
(self)
prnet/ops/pointnet2/pointnet2_stack/voxel_pool_modules.py:60
↓ 1 callersMethodinit_weights
(self)
prnet/ops/pointnet2/pointnet2_stack/pointnet2_modules.py:68
↓ 1 callersMethodinit_weights
(self)
prnet/ops/pointnet2/pointnet2_stack/pointnet2_modules.py:299
↓ 1 callersMethodinit_weights
(self, pretrained=True)
prnet/models/backbones_2d/resnet.py:190
↓ 1 callersMethodinterpolate_from_bev_features
Args: keypoints: (N1 + N2 + ..., 4) bev_features: (B, C, H, W) batch_size: bev_stride:
prnet/models/backbones_3d/pfe/voxel_set_abstraction.py:176
↓ 1 callersFunctioninterpolate_ins_poses
Interpolate poses from INS. Args: ins_path (str): path to file containing poses from INS. pose_timestamps (list[int]): UNIX times
prnet/datasets/oxford/python/interpolate_poses.py:63
↓ 1 callersFunctioninterpolate_vo_poses
Interpolate poses from visual odometry. Args: vo_path (str): path to file containing relative poses from visual odometry. pose_ti
prnet/datasets/oxford/python/interpolate_poses.py:22
↓ 1 callersFunctionlast_conv
(in_channels, out_channels)
prnet/models/localizer/deformable_fusion.py:437
↓ 1 callersFunctionlast_conv
(in_channels, out_channels)
prnet/models/localizer/fusion.py:430
↓ 1 callersFunctionlast_conv
(in_channels, out_channels)
prnet/models/localizer/netvlad.py:188
↓ 1 callersMethodlidar_to_img
:param pts_lidar: (N, 3) :return pts_img: (N, 2)
prnet/utils/calibration_kitti.py:86
↓ 1 callersMethodlidar_to_rect
:param pts_lidar: (N, 3) :return pts_rect: (N, 3)
prnet/utils/calibration_kitti.py:65
↓ 1 callersFunctionload_im_file_for_generate
(filename)
prnet/datasets/nclt/project_vel_to_cam_rot.py:153
↓ 1 callersFunctionload_image
Loads and rectifies an image from file. Args: image_path (str): path to an image from the dataset. model (camera_model.CameraMode
prnet/datasets/oxford/image_preprocess.py:140
↓ 1 callersFunctionload_radar
Decode a single Oxford Radar RobotCar Dataset radar example Args: example_path (AnyStr): Oxford Radar RobotCar Dataset Example png Ret
prnet/datasets/oxford/python/radar.py:20
↓ 1 callersFunctionload_vel_hits
(filename)
prnet/datasets/nclt/project_vel_to_cam_rot.py:41
↓ 1 callersFunctionmain
()
prnet/datasets/oxford/python/play_velodyne.py:47
↓ 1 callersFunctionmain
()
prnet/datasets/nclt/image_preprocess.py:63
↓ 1 callersFunctionmain
(args)
prnet/datasets/nclt/project_vel_to_cam_rot.py:161
↓ 1 callersFunctionmake_dataloaders
Create training and validation dataloaders that return groups of k=2 similar elements :param train_params: :param model_params: :retu
prnet/datasets/dataset_utils.py:173
↓ 1 callersFunctionmake_datasets
(params: TrainingParams, validation=True)
prnet/datasets/dataset_utils.py:16
↓ 1 callersFunctionmake_losses
(params: TrainingParams)
prnet/models/loss.py:9
↓ 1 callersFunctionmerge_intrinsics
(fx, fy, x0, y0)
prnet/utils/vox_utils/geom.py:104
↓ 1 callersFunctionmerge_lrtlist
(lenlist, rtlist)
prnet/utils/vox_utils/geom.py:139
↓ 1 callersFunctionmerge_rt
(r, t)
prnet/utils/vox_utils/geom.py:72
↓ 1 callersFunctionmeshgrid3d
(B, Z, Y, X, stack=False, norm=False, device='cuda')
prnet/utils/vox_utils/basic.py:100
↓ 1 callersMethodmine
(self, embeddings, positives_mask, negatives_mask)
prnet/models/loss.py:98
↓ 1 callersMethodmodel2eval
(self, model)
tools/evaluate.py:101
↓ 1 callersMethodmodel2eval
(self, models)
tools/evaluate.py:130
↓ 1 callersMethodnetwork_initialization
(self, in_channels, out_channels, D)
prnet/models/backbones_2d/resnet.py:18
↓ 1 callersFunctionnormalize
(d)
prnet/utils/vox_utils/basic.py:48
↓ 1 callersFunctionnormalize_grid2d
(grid_y, grid_x, Y, X, clamp_extreme=True)
prnet/utils/vox_utils/basic.py:138
↓ 1 callersFunctionnormalize_grid3d
(grid_z, grid_y, grid_x, Z, Y, X, clamp_extreme=True)
prnet/utils/vox_utils/basic.py:149
↓ 1 callersFunctionnormalize_single
(d)
prnet/utils/vox_utils/basic.py:41
↓ 1 callersMethodone_hot
(self, index, classes)
prnet/models/backbones_3d/focal_sparse_conv/focal_sparse_utils.py:14
↓ 1 callersFunctiononed2inferno
(d, norm=True)
prnet/utils/vox_utils/improc.py:54
↓ 1 callersFunctionop_map
(op)
prnet/utils/commu_utils.py:150
↓ 1 callersFunctionpixels2camera
(x,y,z,fx,fy,x0,y0)
prnet/utils/vox_utils/geom.py:245
↓ 1 callersMethodpreprocess
Preprocess images Args: images: (N, 3, H, W), Input images Return x: (N, 3, H, W), Preprocessed image
prnet/models/backbones_3d/vfe/image_vfe_modules/ffn/ddn/ddn_template.py:143
↓ 1 callersMethodprint_info
(self)
prnet/models/localizer/hownet.py:118
↓ 1 callersFunctionprint_stats
(stats, phase)
tools/trainer.py:23
↓ 1 callersMethodproject
Projects a pointcloud into the camera using a pinhole camera model. Args: xyz (:obj: `numpy.ndarray`): 3xn array, where each colu
prnet/datasets/oxford/python/camera_model.py:54
↓ 1 callersFunctionproject_vel_to_cam
(hits, cam_num)
prnet/datasets/nclt/project_vel_to_cam_rot.py:107
↓ 1 callersFunctionradar_polar_to_cartesian
Convert a polar radar scan to cartesian. Args: azimuths (np.ndarray): Rotation for each polar radar azimuth (radians) fft_data (np
prnet/datasets/oxford/python/radar.py:45
← previousnext →201–300 of 910, ranked by callers