MCPcopy Create free account

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

Functions725 in github.com/BAI-Yeqi/OpenPCSeg

↓ 561 callersMethodsize
Return the size of the point cloud.
pcseg/data/dataset/semantickitti/laserscan.py:73
↓ 44 callersMethodupdate
(self, val, n=1)
tools/utils/common/common_utils.py:262
↓ 13 callersMethodrun_layer
(self, x, layer, skips, os)
pcseg/model/segmentor/range/rangenet/module/darknet.py:140
↓ 12 callersFunctionaug_points
( xyz: np.array, if_flip: bool = False, if_scale: bool = False, scale_axis: str = 'xyz', s
tools/utils/common/seg_utils.py:43
↓ 11 callersFunctionlistify
Make `p` listy and the same length as `q`.
pcseg/optim/fastai_optim.py:80
↓ 10 callersMethod__len__
(self)
pcseg/data/dataset/base.py:94
↓ 8 callersMethod__init__
(self, model_cfgs, num_class)
pcseg/model/segmentor/fusion/rpvnet/rpvnet.py:430
↓ 8 callersMethod_make_layer
(self, block, out_channels, num_block, stride=1, if_dist=False)
pcseg/model/segmentor/fusion/rpvnet/rpvnet.py:422
↓ 8 callersMethod_make_layer
(self, block, out_channels, num_block, stride=1, if_dist=False)
pcseg/model/segmentor/fusion/spvcnn/spvcnn.py:381
↓ 8 callersMethod_make_layer
(self, block, out_channels, num_block, stride=1, if_dist=False)
pcseg/model/segmentor/voxel/minkunet/minkunet.py:364
↓ 8 callersFunctionbuild_dataloader
( data_cfgs, modality: str, batch_size: int, dist: bool = False, root_path: str = None,
pcseg/data/__init__.py:46
↓ 7 callersMethodset_val
Set `val` inside the optimizer dictionary at `key`.
pcseg/optim/fastai_optim.py:223
↓ 7 callersMethodstep
Set weight decay and step optimizer.
pcseg/optim/fastai_optim.py:135
↓ 6 callersMethod__init__
Args: code_weights: (#codes) float list if not None. Code-wise weights.
tools/utils/train/loss.py:198
↓ 6 callersMethod__init__
(self, params={'knn': 5, 'search': 5, 'sigma': 1.0, 'cutoff': 1.0}, nclasses=20)
pcseg/model/segmentor/range/utils.py:292
↓ 6 callersMethod__init__
(self, params={'knn': 5, 'search': 5, 'sigma': 1.0, 'cutoff': 1.0}, nclasses=20)
pcseg/model/segmentor/range/fidnet/model/semantic/utils.py:292
↓ 6 callersFunctionmean
nanmean compatible with generators.
tools/utils/common/lovasz_losses.py:305
↓ 5 callersMethod_make_dec_layer
(self, block, planes, bn_d=0.1, stride=2)
pcseg/model/segmentor/range/rangenet/module/darknet.py:232
↓ 5 callersMethod_make_enc_layer
(self, block, planes, blocks, stride, bn_d=0.1)
pcseg/model/segmentor/range/rangenet/module/darknet.py:125
↓ 5 callersFunctionconv1x3
(in_planes: int, out_planes: int, stride: int = 1, indice_key: str = None)
pcseg/model/segmentor/voxel/cylinder3d/cylinder_ts.py:51
↓ 5 callersFunctionconv3x1
(in_planes: int, out_planes: int, stride: int = 1, indice_key: str = None)
pcseg/model/segmentor/voxel/cylinder3d/cylinder_ts.py:57
↓ 5 callersFunctionload_data_to_gpu
(batch_dict)
pcseg/model/__init__.py:17
↓ 5 callersFunctionmodel_state_to_cpu
(model_state)
tools/utils/train_utils.py:138
↓ 5 callersMethodopen_scan
Open raw scan and fill in attributes.
pcseg/data/dataset/semantickitti/laserscan.py:82
↓ 5 callersMethodread_val
Read a hyperparameter `key` in the optimizer dictionary.
pcseg/optim/fastai_optim.py:231
↓ 5 callersMethodrun_layer
(self, x, layer, skips, os)
pcseg/model/segmentor/range/rangenet/module/darknet.py:251
↓ 5 callersMethodtrain
(self)
train.py:495
↓ 4 callersMethod__init__
( self, model_cfgs, num_class, )
pcseg/model/segmentor/fusion/spvcnn/spvcnn.py:190
↓ 4 callersMethod__init__
( self, model_cfgs, num_class: int, )
pcseg/model/segmentor/range/fidnet/model/semantic/fidnet.py:10
↓ 4 callersMethod__init__
(self, in_filters: int, out_filters: int, kernel_size: tuple = (3, 3, 3), indi
pcseg/model/segmentor/voxel/cylinder3d/cylinder_ts.py:254
↓ 4 callersMethod__init__
( self, model_cfgs, num_class: int, )
pcseg/model/segmentor/voxel/minkunet/minkunet.py:187
↓ 4 callersMethod_make_layer
(self, block, planes, blocks, stride=1, dilate=False)
pcseg/model/segmentor/range/cenet/model/semantic/cenet.py:178
↓ 4 callersMethod_make_layer
(self, block, planes, blocks, stride=1, dilate=False)
pcseg/model/segmentor/range/fidnet/model/semantic/fidnet.py:255
↓ 4 callersFunctionconv1x1
1x1 convolution
pcseg/model/segmentor/range/fidnet/model/semantic/fidnet.py:321
↓ 4 callersMethodcreate
Create an `optim.Optimizer` from `opt_func` with `lr`. Set lr on `layer_groups`.
pcseg/optim/fastai_optim.py:115
↓ 4 callersFunctionflatten_probas
Flattens predictions in the batch
tools/utils/common/lovasz_losses.py:207
↓ 4 callersMethodforward
Arguments: - strategy: MixTeacher strategies. - image: original image, size: [6, H, W]. - label: original
pcseg/data/dataset/semantickitti/semantickitti_rv.py:366
↓ 4 callersMethodget_weight
()
pcseg/model/segmentor/range/fidnet/model/semantic/utils.py:346
↓ 4 callersFunctionrange_to_point
convert 2d range feature map to points feature
pcseg/model/segmentor/fusion/rpvnet/rpvnet.py:49
↓ 4 callersMethodreset
Reset scan members.
pcseg/data/dataset/semantickitti/laserscan.py:45
↓ 4 callersMethodtrain
(self)
infer.py:458
↓ 4 callersFunctionvoxel_to_point
(x, z, nearest=False)
pcseg/model/segmentor/fusion/rpvnet/utils.py:69
↓ 4 callersFunctionvoxel_to_point
(x, z, nearest=False)
pcseg/model/segmentor/fusion/spvcnn/utils.py:69
↓ 4 callersFunctionvoxel_to_point
(x, z, nearest=False)
pcseg/model/segmentor/voxel/minkunet/utils.py:69
↓ 4 callersMethodzero_grad
Clear optimizer gradients.
pcseg/optim/fastai_optim.py:154
↓ 3 callersMethod__init__
(self, in_filters, out_filters, dropout_rate, drop_out=True)
pcseg/model/segmentor/range/salsanext/model/semantic/salsanext.py:118
↓ 3 callersMethod__init__
( self, model_cfgs, num_class: int, )
pcseg/model/segmentor/range/cenet/model/semantic/cenet.py:126
↓ 3 callersMethodbackward
(ctx, top_grad)
pcseg/model/segmentor/fusion/rpvnet/range_lib/range_utils/nn/functional/denselize.py:25
↓ 3 callersFunctioncheck_numpy_to_torch
(x)
tools/utils/common/common_utils.py:20
↓ 3 callersFunctionconv3x3
3x3 convolution with padding
pcseg/model/segmentor/range/fidnet/model/semantic/fidnet.py:325
↓ 3 callersMethodevaluate
(self, dataloader, prefix)
train.py:408
↓ 3 callersMethodevaluate
(self, dataloader, prefix)
infer.py:412
↓ 3 callersFunctionflatten_binary_scores
Flattens predictions in the batch (binary case) Remove labels equal to 'ignore'
tools/utils/common/lovasz_losses.py:119
↓ 3 callersFunctionget_world_size
()
tools/utils/common/commu_utils.py:17
↓ 3 callersMethodload_params
(self, model_state_disk, strict=False)
pcseg/model/segmentor/base_segmentors.py:16
↓ 3 callersMethodopen_label
Open raw scan and fill in attributes.
pcseg/data/dataset/semantickitti/laserscan.py:309
↓ 3 callersFunctionpoint_to_range
args: pf: N x C point features pxpy: N x 3 batch_id,px,py px range: -1~1 py range: -1~1 b: batch_
pcseg/model/segmentor/fusion/rpvnet/rpvnet.py:73
↓ 3 callersFunctionpoint_to_voxel
(x, z)
pcseg/model/segmentor/fusion/rpvnet/utils.py:41
↓ 3 callersFunctionpoint_to_voxel
(x, z)
pcseg/model/segmentor/fusion/spvcnn/utils.py:41
↓ 3 callersMethodresample
(self)
pcseg/data/dataset/waymo/waymo.py:72
↓ 3 callersMethodshow
Visualize the points cloud. Args: save_path (str, optional): path to save image. Default: None.
tools/scripts/vis_waymo.py:174
↓ 3 callersFunctionsplit_bn_bias
Split the layers in `layer_groups` into batchnorm (`bn_types`) and non-batchnorm groups.
pcseg/optim/fastai_optim.py:16
↓ 2 callersMethod__init__
(self, stub_skips, OS=32, feature_depth=1024)
pcseg/model/segmentor/range/rangenet/module/darknet.py:184
↓ 2 callersFunction_draw_points
Draw points on visualizer. Args: points (numpy.array | torch.tensor, shape=[N, 3+C]): points to visualize. vis (:obj:`
tools/scripts/vis_waymo.py:38
↓ 2 callersFunction_draw_points
Draw points on visualizer. Args: points (numpy.array | torch.tensor, shape=[N, 3+C]): points to visualize. vis (:obj:`
tools/visualizer/vis_SemanticKITTI.py:14
↓ 2 callersMethod_get_group_pred
(self, cls_score, apply_activation_func=False)
pcseg/loss/group_softmax.py:131
↓ 2 callersMethod_get_group_pred
(self, cls_score, apply_activation_func=False)
pcseg/loss/group_softmax_fgbg_2.py:131
↓ 2 callersMethodaddBatch
(self, x, y)
pcseg/model/segmentor/range/fidnet/model/semantic/np_ioueval.py:27
↓ 2 callersFunctionbuild_network
(model_cfgs, num_class)
pcseg/model/__init__.py:10
↓ 2 callersFunctionbuild_optimizer
(model, optim_cfg)
pcseg/optim/__init__.py:13
↓ 2 callersFunctionbuild_scheduler
(optimizer, total_iters_each_epoch, total_epochs, optim_cfg)
pcseg/optim/__init__.py:111
↓ 2 callersFunctioncfg_from_list
Set config keys via list (e.g., from command line).
tools/utils/train/config.py:15
↓ 2 callersFunctioncfg_from_yaml_file
(cfg_file, config)
tools/utils/train/config.py:70
↓ 2 callersMethodcol1row2
(self, img, lbl, msk, img_aux, lbl_aux, msk_aux)
pcseg/model/segmentor/range/utils.py:1968
↓ 2 callersMethodcol1row2
(self, img, lbl, msk, img_aux, lbl_aux, msk_aux)
pcseg/model/segmentor/range/fidnet/model/semantic/utils.py:1968
↓ 2 callersFunctionconv1x1
( in_planes: int, out_planes: int, stride: int = 1, )
pcseg/model/segmentor/range/cenet/model/semantic/cenet.py:19
↓ 2 callersFunctionconv3x3
( in_planes: int, out_planes: int, stride: int = 1, groups: int = 1, dilation: int = 1, )
pcseg/model/segmentor/range/cenet/model/semantic/cenet.py:8
↓ 2 callersFunctionconv3x3
(in_planes: int, out_planes: int, stride: int = 1, indice_key: str = None)
pcseg/model/segmentor/voxel/cylinder3d/cylinder_ts.py:63
↓ 2 callersFunctionconvert_range_image_to_point_cloud
Modified from the codes of Waymo Open Dataset. Convert range images to point cloud. Args: frame: open dataset frame range
pcseg/utils/waymo_utils.py:85
↓ 2 callersFunctionconvert_range_image_to_point_cloud_labels
Convert segmentation labels from range images to point clouds. Args: frame: open dataset frame range_images: A dict of {laser_name, [
pcseg/data/dataset/waymo/scripts/preprocess_waymo_data.py:23
↓ 2 callersFunctionflatten_probas
Flattens predictions in the batch
pcseg/model/segmentor/range/utils.py:489
↓ 2 callersFunctionflatten_probas
Flattens predictions in the batch
pcseg/model/segmentor/range/fidnet/model/semantic/utils.py:489
↓ 2 callersMethodgenerate_label
(self,semantic_label)
pcseg/data/dataset/semantickitti/semantickitti_rv.py:330
↓ 2 callersFunctiongenerate_labels
(frame, pose)
pcseg/utils/waymo_utils.py:31
↓ 2 callersMethodgetIoU
(self)
pcseg/model/segmentor/range/fidnet/model/semantic/np_ioueval.py:52
↓ 2 callersMethodgetStats
(self)
pcseg/model/segmentor/range/torch_ioueval.py:55
↓ 2 callersMethodgetStats
(self)
pcseg/model/segmentor/range/np_ioueval.py:41
↓ 2 callersMethodgetStats
(self)
pcseg/model/segmentor/range/fidnet/model/semantic/torch_ioueval.py:55
↓ 2 callersMethodgetStats
(self)
pcseg/model/segmentor/range/fidnet/model/semantic/np_ioueval.py:41
↓ 2 callersMethodget_single_sample
Generates one sample of data
pcseg/data/dataset/semantickitti/semantickitti_cylinder.py:104
↓ 2 callersMethodget_single_sample
Generates one sample of data
pcseg/data/dataset/semantickitti/semantickitti_voxel.py:74
↓ 2 callersMethodget_single_sample
Generates one sample of data
pcseg/data/dataset/semantickitti/semantickitti_fusion.py:128
↓ 2 callersMethodget_single_sample
Generates one sample of data
pcseg/data/dataset/waymo/waymo_cylinder.py:95
↓ 2 callersMethodget_single_sample
Generates one sample of data
pcseg/data/dataset/waymo/waymo_fusion.py:68
↓ 2 callersMethodget_single_sample
Generates one sample of data
pcseg/data/dataset/waymo/waymo_voxel.py:79
↓ 2 callersMethodgetacc
(self)
pcseg/model/segmentor/range/fidnet/model/semantic/np_ioueval.py:60
↓ 2 callersMethodis_background_group
(self, group_id)
pcseg/loss/group_softmax.py:245
↓ 2 callersFunctionis_tuple
(x)
pcseg/optim/fastai_optim.py:100
↓ 2 callersMethodload_params_from_file
(self, filename, logger, to_cpu=False)
pcseg/model/segmentor/base_segmentors.py:28
↓ 2 callersFunctionlog_config_to_file
(cfg, pre='cfg', logger=None)
tools/utils/train/config.py:6
next →1–100 of 725, ranked by callers