MCPcopy Create free account

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

Functions725 in github.com/BAI-Yeqi/OpenPCSeg

↓ 2 callersFunctionlovasz_grad
Computes gradient of the Lovasz extension w.r.t sorted errors See Alg. 1 in paper
tools/utils/common/lovasz_losses.py:23
↓ 2 callersFunctionlovasz_hinge_flat
Binary Lovasz hinge loss logits: [P] Variable, logits at each prediction (between -\infty and +\infty) labels: [P] Tensor, binary gro
tools/utils/common/lovasz_losses.py:99
↓ 2 callersFunctionlovasz_softmax_flat
Multi-class Lovasz-Softmax loss probas: [P, C] Variable, class probabilities at each prediction (between 0 and 1) labels: [P] Tensor,
tools/utils/common/lovasz_losses.py:176
↓ 2 callersFunctionlovasz_softmax_flat
Multi-class Lovasz-Softmax loss probas: [P, C] Variable, class probabilities at each prediction (between 0 and 1) labels: [P] Tensor,
pcseg/model/segmentor/range/utils.py:458
↓ 2 callersFunctionlovasz_softmax_flat
Multi-class Lovasz-Softmax loss probas: [P, C] Variable, class probabilities at each prediction (between 0 and 1) labels: [P] Tensor,
pcseg/model/segmentor/range/fidnet/model/semantic/utils.py:458
↓ 2 callersFunctionmean
nanmean compatible with generators.
pcseg/model/segmentor/range/utils.py:404
↓ 2 callersFunctionmean
nanmean compatible with generators.
pcseg/model/segmentor/range/fidnet/model/semantic/utils.py:404
↓ 2 callersMethodmerge_all_iters_to_one_epoch
(self, merge=True, epochs=None)
pcseg/data/dataset/base.py:87
↓ 2 callersMethodnew
Create a new `OptimWrapper` from `self` with another `layer_groups` but the same hyper-parameters.
pcseg/optim/fastai_optim.py:124
↓ 2 callersMethodprepare_input_label_semantic_with_mask
(self, sample)
pcseg/data/dataset/semantickitti/semantickitti_rv.py:284
↓ 2 callersFunctionresample_grid_stacked
:param predictions: NCHW :param pxpy: Nx3 3: batch_idx, px, py :return:
pcseg/model/segmentor/fusion/rpvnet/rpvnet.py:32
↓ 2 callersMethodreset
(self)
pcseg/model/segmentor/range/np_ioueval.py:22
↓ 2 callersMethodreset
(self)
pcseg/model/segmentor/range/fidnet/model/semantic/np_ioueval.py:22
↓ 2 callersMethodresume
(self, filename)
train.py:303
↓ 2 callersMethodresume
(self, filename)
infer.py:303
↓ 2 callersMethodsample_transform
(self, dataset_dict, split_point)
pcseg/data/dataset/semantickitti/semantickitti_rv.py:304
↓ 2 callersMethodset_label
Set points for label not from file but from numpy.
pcseg/data/dataset/semantickitti/laserscan.py:358
↓ 2 callersMethodsmooth_l1_loss
(diff, beta)
tools/utils/train/loss.py:158
↓ 2 callersFunctiontrainable_params
Return list of trainable params in `m`.
pcseg/optim/fastai_optim.py:94
↓ 1 callersFunctionBackbone
ResNet-34 model from "Deep Residual Learning for Image Recognition" <https://arxiv.org/pdf/1512.03385.pdf>
pcseg/model/segmentor/range/fidnet/model/semantic/fidnet.py:329
↓ 1 callersFunctionNN_filter
(depth, pred, k_size=5)
pcseg/model/segmentor/range/utils.py:254
↓ 1 callersFunctionNN_filter
(depth, pred, k_size=5)
pcseg/model/segmentor/range/fidnet/model/semantic/utils.py:254
↓ 1 callersMethodRangePaste
(self, scan, label, mask, scan_, label_, mask_)
pcseg/data/dataset/semantickitti/semantickitti_rv.py:210
↓ 1 callersMethodRangeUnion
(self, scan, label, mask, scan_, label_, mask_)
pcseg/data/dataset/semantickitti/semantickitti_rv.py:197
↓ 1 callersMethod__getitem__
To support a custom dataset, implement this function to load the raw data (and labels), then transform them to the unified normative
pcseg/data/dataset/base.py:97
↓ 1 callersMethod__init__
(self, fai_optimizer, total_step, lr_max, moms, div_factor, pct_start)
pcseg/optim/learning_schedules_fastai.py:61
↓ 1 callersMethod__init__
(self, params, nclasses)
pcseg/model/segmentor/range/rangenet/postproc/CRF.py:79
↓ 1 callersMethod__init__
( self, project: bool = True, H: int = 64, W: int = 512, fov_up: float = 3.0,
pcseg/data/dataset/semantickitti/laserscan.py:8
↓ 1 callersMethod__init__
(self, ignore_index, smooth=1, exponent=2)
pcseg/loss/dice_loss_v1.py:84
↓ 1 callersMethod__init__
(self, ignore_index, smooth=1, exponent=2)
pcseg/loss/dice_loss_v0.py:70
↓ 1 callersFunction_backbone
(arch, block, layers, if_BN, if_remission, if_range, with_normal)
pcseg/model/segmentor/range/fidnet/model/semantic/fidnet.py:317
↓ 1 callersMethod_check_dimensions
(self, predictions: torch.FloatTensor, targets: torch.LongTensor)
pcseg/model/segmentor/range/utils.py:626
↓ 1 callersMethod_check_dimensions
(self, predictions: torch.FloatTensor, targets: torch.LongTensor)
pcseg/model/segmentor/range/fidnet/model/semantic/utils.py:626
↓ 1 callersMethod_dice_loss_binary
Dice loss for one channel binarized input. :param predictions: <torch.FloatTensor: n_samples, 1, H, W>. Predicted scores. :pa
pcseg/model/segmentor/range/utils.py:558
↓ 1 callersMethod_dice_loss_binary
Dice loss for one channel binarized input. :param predictions: <torch.FloatTensor: n_samples, 1, H, W>. Predicted scores. :pa
pcseg/model/segmentor/range/fidnet/model/semantic/utils.py:558
↓ 1 callersMethod_dice_loss_multichannel
Calculate the loss for multichannel predictions. :param predictions: <torch.FloatTensor: n_samples, n_class, H, W>. Predicted scores.
pcseg/model/segmentor/range/utils.py:581
↓ 1 callersMethod_dice_loss_multichannel
Calculate the loss for multichannel predictions. :param predictions: <torch.FloatTensor: n_samples, n_class, H, W>. Predicted scores.
pcseg/model/segmentor/range/fidnet/model/semantic/utils.py:581
↓ 1 callersMethod_forward_impl
(self, x)
pcseg/model/segmentor/range/fidnet/model/semantic/fidnet.py:282
↓ 1 callersFunction_gather_feat
(feat, ind, mask=None)
tools/utils/train/loss.py:404
↓ 1 callersMethod_get_group
(self, version="bg_fg")
pcseg/loss/group_softmax.py:56
↓ 1 callersMethod_get_group_bgfg
(self)
pcseg/loss/group_softmax.py:86
↓ 1 callersMethod_get_group_bgfg
(self)
pcseg/loss/group_softmax_fgbg_2.py:88
↓ 1 callersMethod_prepare_for_label_remapping
(self)
pcseg/loss/group_softmax.py:108
↓ 1 callersMethod_prepare_for_label_remapping
(self)
pcseg/loss/group_softmax_fgbg_2.py:113
↓ 1 callersFunction_reg_loss
Refer to https://github.com/tianweiy/CenterPoint L1 regression loss Args: regr (batch x max_objects x dim) gt_regr (batch
tools/utils/train/loss.py:372
↓ 1 callersMethod_remap_labels
(self, labels)
pcseg/loss/group_softmax.py:147
↓ 1 callersMethod_remap_labels
(self, labels)
pcseg/loss/group_softmax_fgbg_2.py:144
↓ 1 callersMethod_sample_others
(self, label)
pcseg/loss/group_softmax.py:168
↓ 1 callersFunction_transpose_and_gather_feat
(feat, ind)
tools/utils/train/loss.py:415
↓ 1 callersFunctionabsoluteFilePaths
(directory)
pcseg/data/dataset/semantickitti/semantickitti.py:13
↓ 1 callersMethodaddBatch
(self, x, y)
pcseg/model/segmentor/range/np_ioueval.py:27
↓ 1 callersMethodaddBatch
(self, x, y)
pcseg/model/segmentor/range/fidnet/model/semantic/torch_ioueval.py:33
↓ 1 callersFunctionall_gather
Run all_gather on arbitrary picklable data (not necessarily tensors) Args: data: any picklable object Returns: list[data]
tools/utils/common/commu_utils.py:52
↓ 1 callersMethodbuild_loss_funs
(self, model_cfgs)
pcseg/model/segmentor/range/salsanext/model/semantic/salsanext.py:211
↓ 1 callersMethodbuild_loss_funs
(self, model_cfgs)
pcseg/model/segmentor/range/cenet/model/semantic/cenet.py:207
↓ 1 callersMethodbuild_loss_funs
(self, model_cfgs)
pcseg/model/segmentor/range/rangenet/model/semantic/rangenet.py:62
↓ 1 callersMethodbuild_loss_funs
(self, model_cfgs)
pcseg/model/segmentor/range/fidnet/model/semantic/fidnet.py:35
↓ 1 callersFunctionbuild_segmentor
(model_cfgs, num_class)
pcseg/model/segmentor/__init__.py:47
↓ 1 callersFunctioncart2polar
(input_xyz)
pcseg/data/dataset/semantickitti/semantickitti_cylinder.py:19
↓ 1 callersFunctioncart2polar
(input_xyz)
pcseg/data/dataset/waymo/waymo_cylinder.py:12
↓ 1 callersFunctioncheckpoint_state
(model=None, optimizer=None, epoch=None, it=None)
tools/utils/train_utils.py:145
↓ 1 callersFunctionchildren
(m: nn.Module)
pcseg/optim/__init__.py:53
↓ 1 callersMethodcol16row1
(self, img, lbl, img_aux, lbl_aux)
pcseg/model/segmentor/range/utils.py:1801
↓ 1 callersMethodcol16row1
(self, img, lbl, img_aux, lbl_aux)
pcseg/model/segmentor/range/fidnet/model/semantic/utils.py:1801
↓ 1 callersMethodcol1row16
(self, img, lbl, img_aux, lbl_aux)
pcseg/model/segmentor/range/utils.py:957
↓ 1 callersMethodcol1row16
(self, img, lbl, img_aux, lbl_aux)
pcseg/model/segmentor/range/fidnet/model/semantic/utils.py:957
↓ 1 callersMethodcol1row2
(self, img, lbl, msk, img_aux, lbl_aux, msk_aux)
pcseg/data/dataset/semantickitti/semantickitti_rv.py:481
↓ 1 callersMethodcol1row3
(self, img, lbl, img_aux, lbl_aux)
pcseg/model/segmentor/range/utils.py:851
↓ 1 callersMethodcol1row3
(self, img, lbl, msk, img_aux, lbl_aux, msk_aux)
pcseg/model/segmentor/range/utils.py:1990
↓ 1 callersMethodcol1row3
(self, img, lbl, img_aux, lbl_aux)
pcseg/model/segmentor/range/fidnet/model/semantic/utils.py:851
↓ 1 callersMethodcol1row3
(self, img, lbl, msk, img_aux, lbl_aux, msk_aux)
pcseg/model/segmentor/range/fidnet/model/semantic/utils.py:1990
↓ 1 callersMethodcol1row3
(self, img, lbl, msk, img_aux, lbl_aux, msk_aux)
pcseg/data/dataset/semantickitti/semantickitti_rv.py:503
↓ 1 callersMethodcol1row4
(self, img, lbl, img_aux, lbl_aux)
pcseg/model/segmentor/range/utils.py:870
↓ 1 callersMethodcol1row4
(self, img, lbl, msk, img_aux, lbl_aux, msk_aux)
pcseg/model/segmentor/range/utils.py:2013
↓ 1 callersMethodcol1row4
(self, img, lbl, img_aux, lbl_aux)
pcseg/model/segmentor/range/fidnet/model/semantic/utils.py:870
↓ 1 callersMethodcol1row4
(self, img, lbl, msk, img_aux, lbl_aux, msk_aux)
pcseg/model/segmentor/range/fidnet/model/semantic/utils.py:2013
↓ 1 callersMethodcol1row4
(self, img, lbl, msk, img_aux, lbl_aux, msk_aux)
pcseg/data/dataset/semantickitti/semantickitti_rv.py:526
↓ 1 callersMethodcol1row5
(self, img, lbl, img_aux, lbl_aux)
pcseg/model/segmentor/range/utils.py:890
↓ 1 callersMethodcol1row5
(self, img, lbl, msk, img_aux, lbl_aux, msk_aux)
pcseg/model/segmentor/range/utils.py:2037
↓ 1 callersMethodcol1row5
(self, img, lbl, img_aux, lbl_aux)
pcseg/model/segmentor/range/fidnet/model/semantic/utils.py:890
↓ 1 callersMethodcol1row5
(self, img, lbl, msk, img_aux, lbl_aux, msk_aux)
pcseg/model/segmentor/range/fidnet/model/semantic/utils.py:2037
↓ 1 callersMethodcol1row5
(self, img, lbl, msk, img_aux, lbl_aux, msk_aux)
pcseg/data/dataset/semantickitti/semantickitti_rv.py:550
↓ 1 callersMethodcol1row6
(self, img, lbl, img_aux, lbl_aux)
pcseg/model/segmentor/range/utils.py:911
↓ 1 callersMethodcol1row6
(self, img, lbl, msk, img_aux, lbl_aux, msk_aux)
pcseg/model/segmentor/range/utils.py:2062
↓ 1 callersMethodcol1row6
(self, img, lbl, img_aux, lbl_aux)
pcseg/model/segmentor/range/fidnet/model/semantic/utils.py:911
↓ 1 callersMethodcol1row6
(self, img, lbl, msk, img_aux, lbl_aux, msk_aux)
pcseg/model/segmentor/range/fidnet/model/semantic/utils.py:2062
↓ 1 callersMethodcol1row6
(self, img, lbl, msk, img_aux, lbl_aux, msk_aux)
pcseg/data/dataset/semantickitti/semantickitti_rv.py:575
↓ 1 callersMethodcol1row8
(self, img, lbl, img_aux, lbl_aux)
pcseg/model/segmentor/range/utils.py:933
↓ 1 callersMethodcol1row8
(self, img, lbl, img_aux, lbl_aux)
pcseg/model/segmentor/range/fidnet/model/semantic/utils.py:933
↓ 1 callersMethodcol2row1
(self, img, lbl, img_aux, lbl_aux)
pcseg/model/segmentor/range/utils.py:995
↓ 1 callersMethodcol2row1
(self, img, lbl, msk, img_aux, lbl_aux, msk_aux)
pcseg/model/segmentor/range/utils.py:2088
↓ 1 callersMethodcol2row1
(self, img, lbl, img_aux, lbl_aux)
pcseg/model/segmentor/range/fidnet/model/semantic/utils.py:995
↓ 1 callersMethodcol2row1
(self, img, lbl, msk, img_aux, lbl_aux, msk_aux)
pcseg/model/segmentor/range/fidnet/model/semantic/utils.py:2088
↓ 1 callersMethodcol2row1
(self, img, lbl, msk, img_aux, lbl_aux, msk_aux)
pcseg/data/dataset/semantickitti/semantickitti_rv.py:601
↓ 1 callersMethodcol2row2
(self, img, lbl, img_aux, lbl_aux)
pcseg/model/segmentor/range/utils.py:1013
↓ 1 callersMethodcol2row2
(self, img, lbl, msk, img_aux, lbl_aux, msk_aux)
pcseg/model/segmentor/range/utils.py:2110
↓ 1 callersMethodcol2row2
(self, img, lbl, img_aux, lbl_aux)
pcseg/model/segmentor/range/fidnet/model/semantic/utils.py:1013
↓ 1 callersMethodcol2row2
(self, img, lbl, msk, img_aux, lbl_aux, msk_aux)
pcseg/model/segmentor/range/fidnet/model/semantic/utils.py:2110
↓ 1 callersMethodcol2row2
(self, img, lbl, msk, img_aux, lbl_aux, msk_aux)
pcseg/data/dataset/semantickitti/semantickitti_rv.py:623
↓ 1 callersMethodcol2row3
(self, img, lbl, img_aux, lbl_aux)
pcseg/model/segmentor/range/utils.py:1042
← previousnext →101–200 of 725, ranked by callers