MCPcopy Create free account

hub / github.com/BAI-Yeqi/OpenPCSeg / functions

Functions725 in github.com/BAI-Yeqi/OpenPCSeg

↓ 1 callersMethodforward
(self, batch_dict, return_logit=False, return_tta=False)
pcseg/model/segmentor/fusion/spvcnn/spvcnn.py:399
↓ 1 callersMethodforward
(self, batch_dict, ensemble=False)
pcseg/model/segmentor/voxel/cylinder3d/cylinder_ts.py:514
↓ 1 callersMethodforward
(self, batch_dict, return_logit=False, return_tta=False)
pcseg/model/segmentor/voxel/minkunet/minkunet.py:385
↓ 1 callersMethodgetIoU
(self)
pcseg/model/segmentor/range/np_ioueval.py:52
↓ 1 callersMethodgetIoU
(self)
pcseg/model/segmentor/range/fidnet/model/semantic/torch_ioueval.py:66
↓ 1 callersFunctionget_dist_info
(return_gpu_per_machine=False)
tools/utils/common/common_utils.py:178
↓ 1 callersFunctionget_gaussian_kernel
(kernel_size=3, sigma=2, channels=1)
pcseg/model/segmentor/range/utils.py:275
↓ 1 callersFunctionget_gaussian_kernel
(kernel_size=3, sigma=2, channels=1)
pcseg/model/segmentor/range/rangenet/postproc/KNN.py:11
↓ 1 callersFunctionget_gaussian_kernel
(kernel_size=3, sigma=2, channels=1)
pcseg/model/segmentor/range/fidnet/model/semantic/utils.py:275
↓ 1 callersFunctionget_git_commit_number
()
setup.py:6
↓ 1 callersMethodget_kitti_points_ringID
(self, points)
pcseg/data/dataset/semantickitti/semantickitti.py:86
↓ 1 callersMethodget_last_depth
(self)
pcseg/model/segmentor/range/rangenet/module/darknet.py:273
↓ 1 callersFunctionget_master
Return two lists, one for the model parameters in FP16 and one for the master parameters in FP32.
pcseg/optim/fastai_optim.py:30
↓ 1 callersFunctionget_n_params
(model)
train.py:25
↓ 1 callersFunctionget_n_params
(model)
infer.py:25
↓ 1 callersFunctionget_point_cloud_with_label
(frame, range_images, camera_projections, segmentation_labels, range_image_top_pose)
pcseg/data/dataset/waymo/scripts/preprocess_waymo_data.py:61
↓ 1 callersMethodget_range_image
:param points: xyzir :return: depth_image, reflectivity_image, px(-1~1), py(-1~1)
pcseg/data/dataset/semantickitti/semantickitti_fusion.py:64
↓ 1 callersFunctionget_rank
()
tools/utils/common/commu_utils.py:25
↓ 1 callersMethodget_training_loss
(self, outputs, voxel_label, batch_dict)
pcseg/model/segmentor/voxel/cylinder3d/cylinder_ts.py:508
↓ 1 callersMethodget_weight
(self, cls_score)
pcseg/loss/eqlv2.py:118
↓ 1 callersMethodgetacc
(self)
pcseg/model/segmentor/range/np_ioueval.py:60
↓ 1 callersMethodgetacc
(self)
pcseg/model/segmentor/range/fidnet/model/semantic/torch_ioueval.py:74
↓ 1 callersMethodinit
(args, cfgs)
train.py:235
↓ 1 callersMethodinit
(args, cfgs)
infer.py:235
↓ 1 callersFunctioninitial_voxelize
(z, init_res, after_res)
pcseg/model/segmentor/fusion/rpvnet/utils.py:11
↓ 1 callersFunctioninitial_voxelize
(z, init_res, after_res)
pcseg/model/segmentor/fusion/spvcnn/utils.py:11
↓ 1 callersFunctioninitial_voxelize
(z, init_res, after_res)
pcseg/model/segmentor/voxel/minkunet/utils.py:11
↓ 1 callersMethodis_background_group
(self, group_id)
pcseg/loss/group_softmax_fgbg_2.py:227
↓ 1 callersFunctionlasermix_aug
Generates one sample of data
pcseg/data/dataset/semantickitti/LaserMix_semantickitti.py:11
↓ 1 callersFunctionlinear_warmup_with_cosdecay
(cur_step, warmup_steps, total_steps, min_scale=1e-5)
pcseg/optim/__init__.py:73
↓ 1 callersFunctionlinear_warmup_with_stepdecay
(cur_step, warmup_steps, total_steps, decay_steps, decay_scales)
pcseg/optim/__init__.py:89
↓ 1 callersFunctionlovasz_grad
Computes gradient of the Lovasz extension w.r.t sorted errors See Alg. 1 in paper
pcseg/model/segmentor/range/utils.py:425
↓ 1 callersFunctionlovasz_grad
Computes gradient of the Lovasz extension w.r.t sorted errors See Alg. 1 in paper
pcseg/model/segmentor/range/fidnet/model/semantic/utils.py:425
↓ 1 callersFunctionlovasz_softmax
Multi-class Lovasz-Softmax loss probas: [B, C, H, W] Variable, class probabilities at each prediction (between 0 and 1). Inte
pcseg/model/segmentor/range/utils.py:440
↓ 1 callersFunctionlovasz_softmax
Multi-class Lovasz-Softmax loss probas: [B, C, H, W] Variable, class probabilities at each prediction (between 0 and 1). Inte
pcseg/model/segmentor/range/fidnet/model/semantic/utils.py:440
↓ 1 callersFunctionmain
()
train.py:546
↓ 1 callersFunctionmain
()
infer.py:508
↓ 1 callersFunctionmain
()
tools/scripts/unpack_wod_sequence.py:93
↓ 1 callersFunctionmake_one_hot
Convert class index tensor to one hot encoding tensor. Args: input: A tensor of shape [N, 1, *] num_classes: An int of number o
pcseg/loss/dice_loss_v1.py:7
↓ 1 callersFunctionmake_one_hot
Convert class index tensor to one hot encoding tensor. Args: input: A tensor of shape [N, 1, *] num_classes: An int of number o
pcseg/loss/dice_loss_v0.py:7
↓ 1 callersFunctionmaster2model
Copy `master_params` to `model_params`.
pcseg/optim/fastai_optim.py:68
↓ 1 callersFunctionmerge_new_config
(config, new_config)
tools/utils/train/config.py:50
↓ 1 callersFunctionmodel_func
(model, batch_dict)
pcseg/model/__init__.py:36
↓ 1 callersFunctionmodel_g2master_g
Copy the `model_params` gradients to `master_params` for the optimizer step.
pcseg/optim/fastai_optim.py:52
↓ 1 callersFunctionnum_children
(m: nn.Module)
pcseg/optim/__init__.py:56
↓ 1 callersFunctionone_hot
Return One Hot Label
pcseg/model/segmentor/range/utils.py:717
↓ 1 callersFunctionone_hot
Return One Hot Label
pcseg/model/segmentor/range/fidnet/model/semantic/utils.py:717
↓ 1 callersFunctionop_map
(op)
tools/utils/common/commu_utils.py:152
↓ 1 callersFunctionparse_config
()
train.py:55
↓ 1 callersFunctionparse_config
()
infer.py:55
↓ 1 callersFunctionparse_config
()
tools/scripts/unpack_wod_sequence.py:26
↓ 1 callersFunctionparse_config
()
tools/scripts/vis_waymo.py:207
↓ 1 callersFunctionper_class_iu
(hist)
train.py:43
↓ 1 callersFunctionpolarmix
(pts1, labels1, pts2, labels2, alpha, beta, instance_classes, Omega)
pcseg/data/dataset/semantickitti/PolarMix_semantickitti.py:60
↓ 1 callersFunctionpostprocess_fid
(A, pred, depth, device)
pcseg/model/segmentor/range/utils.py:229
↓ 1 callersFunctionpostprocess_fid
(A, pred, depth, device)
pcseg/model/segmentor/range/fidnet/model/semantic/utils.py:229
↓ 1 callersFunctionpostprocess_knn
(A, pred, depth, knn, device)
pcseg/model/segmentor/range/utils.py:209
↓ 1 callersFunctionpostprocess_knn
(A, pred, depth, knn, device)
pcseg/model/segmentor/range/fidnet/model/semantic/utils.py:209
↓ 1 callersMethodread_defaults
Read the values inside the optimizer for the hyper-parameters.
pcseg/optim/fastai_optim.py:214
↓ 1 callersMethodresample
(self)
pcseg/data/dataset/semantickitti/semantickitti.py:83
↓ 1 callersFunctionresample_grid_stacked_forfusion
:param predictions: NCHW :param pxpy: Nx3 3: batch_idx, px, py :return:
pcseg/model/segmentor/fusion/rpvnet/rpvnet.py:54
↓ 1 callersMethodreset
(self)
tools/utils/common/common_utils.py:256
↓ 1 callersMethodreset
(self)
pcseg/model/segmentor/range/torch_ioueval.py:29
↓ 1 callersMethodreset
(self)
pcseg/model/segmentor/range/fidnet/model/semantic/torch_ioueval.py:29
↓ 1 callersMethodreset
Reset scan members.
pcseg/data/dataset/semantickitti/laserscan.py:288
↓ 1 callersFunctionrotate_copy
(pts, labels, instance_classes, Omega)
pcseg/data/dataset/semantickitti/PolarMix_semantickitti.py:33
↓ 1 callersFunctionsave_checkpoint
(state, filename='checkpoint')
tools/utils/train_utils.py:164
↓ 1 callersMethodsave_checkpoint
(self)
train.py:285
↓ 1 callersMethodsave_checkpoint
(self)
infer.py:285
↓ 1 callersFunctionsave_lidar_points
(frame, cur_save_path, use_two_returns=True)
pcseg/utils/waymo_utils.py:194
↓ 1 callersFunctionscatter_point_inds
(indices, point_inds, shape)
tools/utils/common/common_utils.py:224
↓ 1 callersMethodsem_label_transform
(self,raw_label_map)
pcseg/data/dataset/semantickitti/semantickitti_rv.py:323
↓ 1 callersMethodset_points
Set scan attributes (instead of opening from file).
pcseg/data/dataset/semantickitti/laserscan.py:148
↓ 1 callersMethodshow
Visualize the points cloud. Args: save_path (str, optional): path to save image. Default: None.
tools/visualizer/vis_SemanticKITTI.py:170
↓ 1 callersFunctionshow_rawdata
(class_id, pc_path, label_path)
tools/scripts/vis_waymo.py:189
↓ 1 callersFunctionshow_rawdata
(class_id, pc_path)
tools/visualizer/vis_SemanticKITTI.py:184
↓ 1 callersMethodsigmoid_cross_entropy_with_logits
PyTorch Implementation for tf.nn.sigmoid_cross_entropy_with_logits: max(x, 0) - x * z + log(1 + exp(-abs(x))) in https://www.
tools/utils/train/loss.py:83
↓ 1 callersMethodstep
(self, step)
pcseg/optim/learning_schedules_fastai.py:44
↓ 1 callersFunctionswap
(pt1, pt2, start_angle, end_angle, label1, label2)
pcseg/data/dataset/semantickitti/PolarMix_semantickitti.py:9
↓ 1 callersFunctiontorch_unique
(x)
tools/utils/common/seg_utils.py:22
↓ 1 callersFunctiontrain_one_epoch
(model, optimizer, train_loader, model_func, lr_scheduler, accumulated_iter, optim_cfg, ra
tools/utils/train_utils.py:11
↓ 1 callersMethodtrain_one_epoch
(self, tbar, data_cfg)
train.py:320
↓ 1 callersMethodtrain_one_epoch
(self, tbar, data_cfg)
infer.py:324
↓ 1 callersFunctionvoxelize
(z, init_res=None, after_res=None, voxel_type='max')
tools/utils/common/seg_utils.py:172
↓ 1 callersFunctionvoxelize_with_label
(point_coords, point_labels, num_classes)
pcseg/data/dataset/semantickitti/semantickitti_cylinder.py:32
↓ 1 callersFunctionvoxelize_with_label
(point_coords, point_labels, num_classes)
pcseg/data/dataset/waymo/waymo_cylinder.py:25
↓ 1 callersMethodweight_initialization
(self)
pcseg/model/segmentor/fusion/rpvnet/rpvnet.py:626
↓ 1 callersMethodweight_initialization
(self)
pcseg/model/segmentor/fusion/spvcnn/spvcnn.py:389
↓ 1 callersMethodweight_initialization
(self)
pcseg/model/segmentor/voxel/cylinder3d/cylinder_ts.py:128
↓ 1 callersMethodweight_initialization
(self)
pcseg/model/segmentor/voxel/cylinder3d/cylinder_ts.py:218
↓ 1 callersMethodweight_initialization
(self)
pcseg/model/segmentor/voxel/cylinder3d/cylinder_ts.py:305
↓ 1 callersMethodweight_initialization
(self)
pcseg/model/segmentor/voxel/minkunet/minkunet.py:376
↓ 1 callersFunctionwrite_version_to_file
(version, target_file)
setup.py:14
FunctionPYBIND11_MODULE
pcseg/model/segmentor/fusion/rpvnet/range_lib/range_utils/src/rangelib_bindings_gpu.cpp:7
Method__getattr__
(self, k: str)
pcseg/optim/fastai_optim.py:159
Method__getitem__
(self, index)
pcseg/data/dataset/semantickitti/semantickitti_cylinder.py:92
Method__getitem__
(self, index)
pcseg/data/dataset/semantickitti/semantickitti_rv.py:121
Method__getitem__
(self, index)
pcseg/data/dataset/semantickitti/semantickitti_voxel.py:62
Method__getitem__
(self, index)
pcseg/data/dataset/semantickitti/semantickitti.py:98
Method__getitem__
(self, index)
pcseg/data/dataset/semantickitti/semantickitti_fusion.py:116
← previousnext →301–400 of 725, ranked by callers