MCPcopy Create free account

hub / github.com/MaverickPeter/vDiSCO / functions

Functions910 in github.com/MaverickPeter/vDiSCO

Methodis_full
(self)
prnet/utils/vox_utils/misc.py:58
Functionis_main_process
()
prnet/utils/commu_utils.py:31
Functionkeep_arrays_by_name
(gt_names, used_classes)
prnet/utils/common_utils.py:138
Functionlimit_period
(val, offset=0.5, period=np.pi)
prnet/utils/common_utils.py:21
Methodload_pcs
(self, scan_paths)
prnet/datasets/oxford/oxford_raw.py:272
Methodload_pcs
(self, scan_paths)
prnet/datasets/oxford/oxford_raw.py:373
Methodload_pcs
(self, scan_paths)
prnet/datasets/nclt/nclt_raw.py:180
Methodload_pcs
(self, scan_paths)
prnet/datasets/nclt/nclt_raw.py:276
Functionload_road_boundary_mask
(mask_path, model)
prnet/datasets/oxford/python/road_boundary.py:24
Functionload_velodyne_binary
Decode a binary Velodyne example (of the form '<timestamp>.bin') Args: example_path (AnyStr): Oxford Radar RobotCar Dataset binary Velodyn
prnet/datasets/oxford/velodyne.py:33
Functionload_velodyne_raw
Decode a raw Velodyne example. (of the form '<timestamp>.png') Args: example_path (AnyStr): Oxford Radar RobotCar Dataset raw Velodyne exa
prnet/datasets/oxford/velodyne.py:54
Functionm2xyz_ypr
(m)
prnet/utils/data_utils/poses.py:35
Functionmake_open3d_feature
(data, dim, npts)
prnet/utils/data_utils/point_clouds.py:64
Functionmake_open3d_point_cloud
(xyz, color=None)
prnet/utils/data_utils/point_clouds.py:71
Functionmask_points_by_range
(points, limit_range)
prnet/utils/common_utils.py:60
Functionmatmul2
(mat1, mat2)
prnet/utils/vox_utils/basic.py:21
Functionmerge_lrt
(l, rt)
prnet/utils/vox_utils/py.py:15
Functionmerge_results_dist
(result_part, size, tmpdir)
prnet/utils/common_utils.py:211
Functionmerge_rt
(r, t)
prnet/utils/vox_utils/py.py:3
Functionmerge_rtlist
(rlist, tlist)
prnet/utils/vox_utils/geom.py:115
Functionmeshgrid2d
(B, Y, X, stack=False, norm=False, device='cuda')
prnet/utils/vox_utils/basic.py:77
Functionmetrics_mean
(l)
prnet/models/l_utils.py:336
Functionms_deform_attn_backward
prnet/ops/src/ms_deform_attn.h:41
Functionms_deform_attn_cpu_backward
prnet/ops/src/cpu/ms_deform_attn_cpu.cpp:29
Functionms_deform_attn_cpu_forward
prnet/ops/src/cpu/ms_deform_attn_cpu.cpp:17
Functionms_deform_attn_forward
prnet/ops/src/ms_deform_attn.h:20
Functionmulti_classes_nms
Args: cls_scores: (N, num_class) box_preds: (N, 7 + C) nms_config: score_thresh: Returns:
prnet/models/model_utils/model_nms_utils.py:28
Functionneg_loss_cornernet
Refer to https://github.com/tianweiy/CenterPoint. Modified focal loss. Exactly the same as CornerNet. Runs faster and costs a little bit more
prnet/utils/loss_utils.py:207
Functionneg_loss_cornernet
Refer to https://github.com/tianweiy/CenterPoint. Modified focal loss. Exactly the same as CornerNet. Runs faster and costs a little bit more
prnet/models/l_utils.py:211
Functionnormalize_coords
Normalize coordinates of a grid between [-1, 1] Args: coords: (..., 3), Coordinates in grid shape: (3), Grid shape Return
prnet/utils/transform_utils.py:38
Functionopt_n_threads
prnet/ops/pointnet2/pointnet2_batch/src/cuda_utils.h:10
Functionpack_seqdim
(tensor, B)
prnet/utils/vox_utils/basic.py:24
Functionpcawhitenlearn_shrinkage
Learn PCA whitening with shrinkage from given descriptors
prnet/models/localizer/hownet.py:29
Functionpost_act_block
(in_channels, out_channels, kernel_size, indice_key=None, stride=1, padding=0, conv_type='s
prnet/models/backbones_3d/spconv_backbone_focal.py:39
Functionpost_act_block
(in_channels, out_channels, kernel_size, indice_key=None, stride=1, padding=0, conv_type='s
prnet/models/backbones_3d/spconv_backbone.py:8
Functionpost_act_block
(in_channels, out_channels, kernel_size, indice_key=None, stride=1, padding=0, conv_type='s
prnet/models/backbones_3d/spconv_backbone_2d.py:8
Functionpost_act_block_dense
(in_channels, out_channels, kernel_size, stride=1, padding=0, dilation=1, norm_fn=None)
prnet/models/backbones_3d/spconv_backbone_2d.py:30
Functionpreprocess_pointcloud
(pc, remove_zero_points: bool = False, min_x: float = None, max_x: float = None,
prnet/datasets/dataset_utils.py:229
Methodprint
(self)
prnet/utils/params.py:212
Functionprint_
(name, tensor)
prnet/utils/vox_utils/basic.py:5
Methodprint_info
(self)
prnet/models/localizer/localizer_template.py:21
Methodprint_info
(self)
prnet/models/localizer/deformable_fusion.py:414
Methodprint_info
(self)
prnet/models/localizer/dolgnet.py:123
Methodprint_info
(self)
prnet/models/localizer/fusion.py:405
Methodprint_info
(self)
prnet/models/localizer/deformable.py:441
Methodprint_info
(self)
prnet/models/localizer/netvlad.py:169
Functionprint_stats
(name, tensor)
prnet/utils/vox_utils/basic.py:9
Methodproject
Projects a pointcloud into the camera using a pinhole camera model. Args: xyz (:obj: `numpy.ndarray`): 3xn array, where each colu
prnet/datasets/oxford/image_preprocess.py:47
Functionproject_to_image
Project points to image Args: project [torch.tensor(..., 3, 4)]: Projection matrix points [torch.Tensor(..., 3)]: 3D points
prnet/utils/transform_utils.py:14
Functionq2r
(q)
prnet/utils/data_utils/poses.py:5
Functionquery_stacked_local_neighbor_idxs_wrapper_stack
prnet/ops/pointnet2/pointnet2_stack/src/vector_pool.cpp:32
Functionquery_three_nn_by_stacked_local_idxs_wrapper_stack
prnet/ops/pointnet2/pointnet2_stack/src/vector_pool.cpp:75
Functionrandom_rotation
(xyz, angle_range=(-np.pi, np.pi))
prnet/datasets/oxford/utils.py:163
Functionrandom_rotation
(xyz, angle_range=(-np.pi, np.pi))
prnet/datasets/nclt/utils.py:143
Methodread_pc
(self, file_pathname: list, extrinsics_dir=None)
prnet/datasets/oxford/oxford_raw.py:145
Methodread_pc
(self, file_pathname: str, sph=False, extrinsics_dir=None)
prnet/datasets/nclt/nclt_raw.py:123
Methodread_pcim
(self, file_pathname: list, sph=False, extrinsics_dir=None)
prnet/datasets/oxford/oxford_raw.py:111
Methodread_pcim
(self, file_pathname: str, sph=False, extrinsics_dir=None)
prnet/datasets/nclt/nclt_raw.py:99
Methodrect_to_lidar
:param pts_lidar: (N, 3) :return pts_rect: (N, 3)
prnet/utils/calibration_kitti.py:50
Functionreduce_dict
Args: input_dict (dict): all the values will be reduced average (bool): whether to do average or sum Reduce the values in the
prnet/utils/commu_utils.py:114
Functionreduce_masked_mean
(x, mask, dim=None, keepdim=False)
prnet/utils/vox_utils/basic.py:58
Functionrelative_pose
(m1, m2)
prnet/utils/data_utils/poses.py:80
Functionremove_the_blackborder
(image)
prnet/datasets/panorama.py:19
Functionrotate_points_along_z
Args: points: (B, N, 3 + C) angle: (B), angle along z-axis, angle increases x ==> y Returns:
prnet/utils/common_utils.py:35
Functionrotation_on_depth
(depth, rotation)
prnet/datasets/dataset_utils.py:272
Functionsa_create
(name, var)
prnet/utils/common_utils.py:255
Functionsafe_inverse
(a)
prnet/utils/vox_utils/geom.py:10
Functionscale_intrinsics
(K, sx, sy)
prnet/utils/vox_utils/geom.py:87
Functionse3_to_components
Converts an SE3 rotation matrix to linear translation and Euler angles Args: se3: 4x4 transformation matrix Returns: numpy.m
prnet/datasets/oxford/utils.py:363
Functionse3_to_components
Converts an SE3 rotation matrix to linear translation and Euler angles Args: se3: 4x4 transformation matrix Returns: numpy.m
prnet/datasets/oxford/python/transform.py:164
Functionset_bn_momentum
(model, momentum=0.1)
prnet/models/attention/attentions.py:18
Functionset_bn_momentum
(model, momentum=0.1)
prnet/models/localizer/deformable.py:48
Methodset_mask
(fmask)
prnet/datasets/nclt/image_preprocess.py:46
Methodset_properties
(self)
prnet/datasets/oxford/oxford_raw.py:106
Methodset_properties
(self)
prnet/datasets/oxford/oxford_raw.py:141
Methodset_properties
(self)
prnet/datasets/nclt/nclt_raw.py:95
Methodset_properties
(self)
prnet/datasets/nclt/nclt_raw.py:119
Functionset_random_seed
(seed)
prnet/utils/common_utils.py:102
Functionso3_to_quaternion
Converts an SO3 rotation matrix to a quaternion Args: so3: 3x3 rotation matrix Returns: numpy.ndarray: quaternion [w, x, y,
prnet/datasets/oxford/utils.py:304
Functionsplit_rt
(rt)
prnet/utils/vox_utils/geom.py:67
Functionsquared_euclidean_distance
Compute squared Euclidean distance Input: x is Nxd matrix y is Mxd matirx Output: dist is a NxM matrix where dist[i,j] is the
prnet/models/l_utils.py:351
Functionstack_farthest_point_sampling_wrapper
prnet/ops/pointnet2/pointnet2_stack/src/sampling.cpp:37
Functionstrnum
(x)
prnet/utils/vox_utils/basic.py:14
Methodsumm_flow
(self, name, im, clip=0.0, only_return=False, frame_id=None)
prnet/utils/vox_utils/improc.py:274
Methodsumm_occ
(self, name, occ, reduce_axes=[3], bev=False, fro=False, pro=False, frame_id=None, only_return=False)
prnet/utils/vox_utils/improc.py:219
Methodsumm_scalar
(self, name, value)
prnet/utils/vox_utils/improc.py:150
Functionsynchronize
Helper function to synchronize (barrier) among all processes when using distributed training
prnet/utils/commu_utils.py:35
Functiontensor2img
(x)
prnet/datasets/augmentation.py:288
Functionthree_interpolate_grad_wrapper_fast
prnet/ops/pointnet2/pointnet2_batch/src/interpolate.cpp:44
Functionthree_interpolate_grad_wrapper_stack
prnet/ops/pointnet2/pointnet2_stack/src/interpolate.cpp:86
Functionthree_interpolate_wrapper_fast
prnet/ops/pointnet2/pointnet2_batch/src/interpolate.cpp:29
Functionthree_interpolate_wrapper_stack
prnet/ops/pointnet2/pointnet2_stack/src/interpolate.cpp:63
Functionthree_nn_wrapper_fast
prnet/ops/pointnet2/pointnet2_batch/src/interpolate.cpp:18
Functionthree_nn_wrapper_stack
prnet/ops/pointnet2/pointnet2_stack/src/interpolate.cpp:32
Methodto_object
(obj: object, **data)
prnet/models/backbones_3d/spconv_backbone_focal.py:13
Functionunpack_seqdim
(tensor, B)
prnet/utils/vox_utils/basic.py:32
Methodupdate
(self, items)
prnet/utils/vox_utils/misc.py:67
Functionvector_pool_grad_wrapper_stack
prnet/ops/pointnet2/pointnet2_stack/src/vector_pool.cpp:170
Functionvector_pool_wrapper_stack
prnet/ops/pointnet2/pointnet2_stack/src/vector_pool.cpp:113
Functionvelodyne_raw_to_pointcloud
Convert raw Velodyne data (from load_velodyne_raw) into a pointcloud Args: ranges (np.ndarray): Raw Velodyne range readings inten
prnet/datasets/oxford/velodyne.py:85
← previousnext →801–900 of 910, ranked by callers