MCPcopy Create free account

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

Functions725 in github.com/BAI-Yeqi/OpenPCSeg

Methodforward
Warning! Only works for un-batched pointclouds. If they come batched we need to iterate over the batch dimension or do something REAL
pcseg/model/segmentor/range/rangenet/postproc/KNN.py:54
Methodforward
(self, range_label)
pcseg/model/segmentor/range/rangenet/postproc/borderMask.py:129
Methodforward
(self, xyz, softmax, mask)
pcseg/model/segmentor/range/rangenet/postproc/CRF.py:25
Methodforward
(self, input, softmax, mask)
pcseg/model/segmentor/range/rangenet/postproc/CRF.py:112
Methodforward
(self, batch, mask=None)
pcseg/model/segmentor/range/rangenet/model/semantic/rangenet.py:78
Methodforward
(self, batch)
pcseg/model/segmentor/range/fidnet/model/semantic/fidnet.py:50
Methodforward
(self, x)
pcseg/model/segmentor/range/fidnet/model/semantic/fidnet.py:114
Methodforward
(self, x)
pcseg/model/segmentor/range/fidnet/model/semantic/fidnet.py:149
Methodforward
(self, input_tensor)
pcseg/model/segmentor/range/fidnet/model/semantic/fidnet.py:184
Methodforward
(self, x)
pcseg/model/segmentor/range/fidnet/model/semantic/fidnet.py:313
Methodforward
(self, proj_range, unproj_range, proj_argmax, px, py)
pcseg/model/segmentor/range/fidnet/model/semantic/utils.py:300
Methodforward
(self, probas, labels)
pcseg/model/segmentor/range/fidnet/model/semantic/utils.py:516
Methodforward
Forward pass. :param predictions: <torch.FloatTensor: n_samples, C, H, W>. Predicted scores. :param targets: <torch.LongTenso
pcseg/model/segmentor/range/fidnet/model/semantic/utils.py:544
Methodforward
Calculate the loss. :param predictions: <torch.FloatTensor: n_samples, n_class, H, W>. Predicted scores. :param targets: <tor
pcseg/model/segmentor/range/fidnet/model/semantic/utils.py:653
Methodforward
Input: - pred: the output from model (before softmax), shape (N, C, H, W) - gt: ground truth map, shape (N, H, w)
pcseg/model/segmentor/range/fidnet/model/semantic/utils.py:678
Methodforward
Arguments: - strategy: RangeAug strategies. - image: original image, size: [bs, 6, h, w]. - label: origin
pcseg/model/segmentor/range/fidnet/model/semantic/utils.py:745
Methodforward
Arguments: - strategy: MixTeacher strategies. - image: original image, size: [6, H, W]. - label: original
pcseg/model/segmentor/range/fidnet/model/semantic/utils.py:1853
Methodforward
(self, x)
pcseg/model/segmentor/voxel/cylinder3d/cylinder_ts.py:137
Methodforward
(self, x)
pcseg/model/segmentor/voxel/cylinder3d/cylinder_ts.py:227
Methodforward
(self, x, skip)
pcseg/model/segmentor/voxel/cylinder3d/cylinder_ts.py:314
Methodforward
(self, x)
pcseg/model/segmentor/voxel/cylinder3d/cylinder_ts.py:368
Methodforward
(self, input: SparseTensor)
pcseg/model/segmentor/voxel/minkunet/minkunet.py:24
Methodforward
(self, input: SparseTensor)
pcseg/model/segmentor/voxel/minkunet/minkunet.py:28
Methodforward
(self, x)
pcseg/model/segmentor/voxel/minkunet/minkunet.py:53
Methodforward
(self, x)
pcseg/model/segmentor/voxel/minkunet/minkunet.py:79
Methodforward
(self, x)
pcseg/model/segmentor/voxel/minkunet/minkunet.py:127
Methodforward
(self, x)
pcseg/model/segmentor/voxel/minkunet/minkunet.py:181
Methodforward
(self, predict, target)
pcseg/loss/dice_loss_v1.py:45
Methodforward
(self, input, target)
pcseg/loss/dice_loss_v1.py:90
Methodforward
(self, input, target)
pcseg/loss/focalloss.py:17
Methodforward
(self, input, target, xyz=None, offset=None)
pcseg/loss/__init__.py:83
Methodforward
Args: target: points gt labels for one batch xyz: coords of all points -> voxelized xyz : lidar.C[:, :3].float() or l
pcseg/loss/geo_loss.py:18
Methodforward
(self, cls_score, label, weight=None, avg_fact
pcseg/loss/group_softmax.py:195
Methodforward
(self, predict, target)
pcseg/loss/dice_loss_v0.py:45
Methodforward
(self, input, target)
pcseg/loss/dice_loss_v0.py:76
Methodforward
(self, input ,target)
pcseg/loss/wce_loss.py:28
Methodforward
(self, cls_score, label, weight=None, avg_fact
pcseg/loss/eqlv2.py:46
Methodforward
L_exp = w_dice * L_dice + w_cross * L_cross L_dice = E[(-ln(Dice))^{\gamma_{dice}}] E -> mean value Dice_i = [2(pred_score
pcseg/loss/ell_loss.py:79
Methodforward
(self, cls_score, label, focal_loss=False, wei
pcseg/loss/group_softmax_fgbg_2.py:180
Methodforward_ensemble
(self, batch_dict)
pcseg/model/segmentor/fusion/rpvnet/rpvnet.py:751
Methodforward_ensemble
(self, batch_dict)
pcseg/model/segmentor/fusion/spvcnn/spvcnn.py:483
Methodforward_ensemble
(self, batch_dict)
pcseg/model/segmentor/voxel/cylinder3d/cylinder_ts.py:590
Methodforward_ensemble
(self, batch_dict)
pcseg/model/segmentor/voxel/minkunet/minkunet.py:457
Methodgenerate_prediction_dicts
To support a custom dataset, implement this function to receive the predicted results from the model, and then transform the unified
pcseg/data/dataset/base.py:70
Functiongenerate_voxel2pinds
(sparse_tensor)
tools/utils/common/common_utils.py:233
MethodgetIoU
(self)
pcseg/model/segmentor/range/torch_ioueval.py:66
Methodget_activation
(self, cls_score, bgfgweight=False, apply_activation_func=False)
pcseg/loss/group_softmax.py:221
Methodget_activation
(self, cls_score, bgfgweight=False, apply_activation_func=False)
pcseg/loss/eqlv2.py:94
Methodget_activation
(self, cls_score, bgfgweight=False, apply_activation_func=False)
pcseg/loss/group_softmax_fgbg_2.py:206
Methodget_activation_for_group
(self, cls_score)
pcseg/loss/group_softmax.py:237
Methodget_bin_weight
(bin_num)
pcseg/model/segmentor/range/utils.py:369
Methodget_bin_weight
(bin_num)
pcseg/model/segmentor/range/fidnet/model/semantic/utils.py:369
Methodget_channel_num
(self)
pcseg/loss/group_softmax.py:241
Methodget_channel_num
(self, num_classes)
pcseg/loss/eqlv2.py:90
Methodget_channel_num
(self)
pcseg/loss/group_softmax_fgbg_2.py:223
Methodget_confusion
(self)
pcseg/model/segmentor/range/np_ioueval.py:67
Methodget_confusion
(self)
pcseg/model/segmentor/range/fidnet/model/semantic/np_ioueval.py:67
Functionget_corner_loss_lidar
Args: pred_bbox3d: (N, 7) float Tensor. gt_bbox3d: (N, 7) float Tensor. Returns: corner_loss: (N) float Tensor.
tools/utils/train/loss.py:267
Methodget_input_depth
(self)
pcseg/model/segmentor/range/rangenet/module/darknet.py:179
Methodget_last_depth
(self)
pcseg/model/segmentor/range/rangenet/module/darknet.py:176
Methodget_lr
(self)
pcseg/optim/learning_schedules_fastai.py:86
Functionget_n_params
(model)
pcseg/model/segmentor/range/utils.py:376
Functionget_n_params
(model)
pcseg/model/segmentor/range/fidnet/model/semantic/utils.py:376
Functionget_pad_params
Get padding parameters for np.pad function Args: desired_size: int, Desired padded output size cur_size: int, Current size. S
tools/utils/common/common_utils.py:107
Functionget_point_cloud_without_label
(frame, range_images, camera_projections, segmentation_labels, range_image_top_pose)
pcseg/data/dataset/waymo/scripts/preprocess_waymo_data.py:89
Functionget_semantic_segmentation
(sem)
pcseg/model/segmentor/range/utils.py:386
Functionget_semantic_segmentation
(sem)
pcseg/model/segmentor/range/fidnet/model/semantic/utils.py:386
Functionget_voxel_centers
(voxel_coords, downsample_times, voxel_size, point_cloud_range)
tools/utils/common/common_utils.py:72
Methodget_weight
()
pcseg/model/segmentor/range/utils.py:346
Methodgetacc
(self)
pcseg/model/segmentor/range/torch_ioueval.py:74
Functionhinge_jaccard_loss
Multi-class Hinge Jaccard loss probas: [B, C, H, W] Variable, class probabilities at each prediction (between 0 and 1). Inter
tools/utils/common/lovasz_losses.py:264
Functioninit_dist_pytorch
(tcp_port, local_rank, backend='nccl')
tools/utils/common/common_utils.py:160
Functioninit_dist_slurm
modified from https://github.com/open-mmlab/mmdetection Args: tcp_port: backend: Returns:
tools/utils/common/common_utils.py:131
Functioninitial_voxelize_max
(z, init_res, after_res)
pcseg/model/segmentor/voxel/cylinder3d/cylinder_ts.py:24
Functioniou
Array of IoU for each (non ignored) class
tools/utils/common/lovasz_losses.py:58
Functioniou_binary
IoU for foreground class binary: 1 foreground, 0 background
tools/utils/common/lovasz_losses.py:38
Functionis_main_process
()
tools/utils/common/commu_utils.py:33
Functionisnan
(x)
tools/utils/common/lovasz_losses.py:301
Functionisnan
(x)
pcseg/model/segmentor/range/utils.py:400
Functionisnan
(x)
pcseg/model/segmentor/range/fidnet/model/semantic/utils.py:400
Functionjaccard_loss
Something wrong with this loss Multi-class Lovasz-Softmax loss probas: [B, C, H, W] Variable, class probabilities at each prediction (b
tools/utils/common/lovasz_losses.py:236
Functionkeep_arrays_by_name
(gt_names, used_classes)
tools/utils/common/common_utils.py:125
Functionlimit_period
(val, offset=0.5, period=np.pi)
tools/utils/common/common_utils.py:26
Functionlovasz_hinge
Binary Lovasz hinge loss logits: [B, H, W] Variable, logits at each pixel (between -\infty and +\infty) labels: [B, H, W] Tensor, bin
tools/utils/common/lovasz_losses.py:83
Functionlovasz_softmax
Multi-class Lovasz-Softmax loss probas: [B, C, H, W] Variable, class probabilities at each prediction (between 0 and 1). Inte
tools/utils/common/lovasz_losses.py:158
Methodlr
(self)
pcseg/optim/fastai_optim.py:170
Functionmap_count
(pxpy, max_bs, h, w)
pcseg/model/segmentor/fusion/rpvnet/range_lib/range_utils/nn/functional/map_count.py:27
Functionmap_count_forward
pcseg/model/segmentor/fusion/rpvnet/range_lib/range_utils/src/map_count.cpp:5
Functionmap_state_dict
(state_dict, model_dict)
pcseg/model/segmentor/range/utils.py:3190
Functionmap_state_dict
(state_dict, model_dict)
pcseg/model/segmentor/range/fidnet/model/semantic/utils.py:3190
Functionmask_points_by_range
(points, limit_range)
tools/utils/common/common_utils.py:65
Functionmerge_results_dist
(result_part, size, tmpdir)
tools/utils/common/common_utils.py:200
Methodmode
(self)
pcseg/data/dataset/base.py:58
Functionmodel_fn_decorator
()
pcseg/model/__init__.py:33
Methodmom
(self)
pcseg/optim/fastai_optim.py:178
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
tools/utils/train/loss.py:322
Methodnum_classes
(self)
pcseg/model/segmentor/range/torch_ioueval.py:26
Methodnum_classes
(self)
pcseg/model/segmentor/range/np_ioueval.py:19
Methodnum_classes
(self)
pcseg/model/segmentor/range/fidnet/model/semantic/torch_ioueval.py:26
Methodnum_classes
(self)
pcseg/model/segmentor/range/fidnet/model/semantic/np_ioueval.py:19
← previousnext →601–700 of 725, ranked by callers