MCPcopy Create free account

hub / github.com/Jasonkks/PTTR / functions

Functions1,101 in github.com/Jasonkks/PTTR

↓ 56 callersMethod__len__
(self)
OpenPCDet/pcdet/datasets/dataset.py:85
↓ 44 callersMethodupdate
(self, val, n=1)
OpenPCDet/pcdet/utils/common_utils.py:263
↓ 35 callersFunctionmin
OpenPCDet/pcdet/ops/iou3d_nms/src/iou3d_cpu.cpp:30
↓ 34 callersFunctionmax
OpenPCDet/pcdet/ops/iou3d_nms/src/iou3d_cpu.cpp:34
↓ 24 callersMethodrotate
Rotates box. :param quaternion: <Quaternion>. Rotation to apply. :return: <None>.
data_classes.py:210
↓ 23 callersMethodtranslate
Applies a translation. :param x: <np.float: 3, 1>. Translation in x, y, z direction. :return: <None>.
data_classes.py:202
↓ 20 callersMethodget_loss
(self)
OpenPCDet/pcdet/models/dense_heads/center_head.py:225
↓ 16 callersMethodsigmoid
(self, x)
OpenPCDet/pcdet/models/dense_heads/center_head.py:221
↓ 15 callersFunctionprint_str
(value, *arg, sstream=None)
OpenPCDet/pcdet/datasets/kitti/kitti_object_eval_python/eval.py:570
↓ 13 callersFunctionreplace_feature
(out, new_features)
OpenPCDet/pcdet/utils/spconv_utils.py:28
↓ 12 callersMethodupdate
(self, val, n=1)
metrics.py:19
↓ 11 callersFunctionlistify
Make `p` listy and the same length as `q`.
OpenPCDet/tools/train_utils/optimization/fastai_optim.py:80
↓ 10 callersMethodbuild_networks
(self)
OpenPCDet/pcdet/models/detectors/detector3d_template.py:35
↓ 10 callersMethodset
OpenPCDet/pcdet/ops/iou3d_nms/src/iou3d_cpu.cpp:46
↓ 9 callersFunctionget_points_in_box
(points, gt_box)
OpenPCDet/pcdet/datasets/augmentor/augmentor_utils.py:474
↓ 9 callersMethodpost_processing
(self, batch_dict)
OpenPCDet/pcdet/models/detectors/centerpoint.py:36
↓ 8 callersMethodcorners
Returns the bounding box corners. :param wlh_factor: <float>. Multiply w, l, h by a factor to inflate or deflate the box. :re
data_classes.py:227
↓ 7 callersMethod__init__
( self, in_size: int, out_size: int, *, activation
pointnet2/_ext-src/pointnet2_3DSSD/pytorch_utils.py:202
↓ 7 callersMethodgetitem
(self, index)
Dataset.py:346
↓ 7 callersMethodmake_fc_layers
(self, input_channels, output_channels, fc_list)
OpenPCDet/pcdet/models/roi_heads/roi_head_template.py:29
↓ 7 callersMethodnbr_points
Returns the number of points. :return: <int>. Number of points.
data_classes.py:50
↓ 7 callersMethodsample
(self)
searchspace.py:11
↓ 7 callersMethodset_val
Set `val` inside the optimizer dictionary at `key`.
OpenPCDet/tools/train_utils/optimization/fastai_optim.py:223
↓ 7 callersMethodstep
Set weight decay and step optimizer.
OpenPCDet/tools/train_utils/optimization/fastai_optim.py:135
↓ 6 callersMethod__init__
(self, d_model=512, nhead=1, num_layers=1, dim_feedforward=2048, activation="relu")
pointnet2/models/transformer.py:30
↓ 6 callersMethod__len__
(self)
OpenPCDet/tools/demo.py:43
↓ 6 callersFunction_extend_matrix
(mat)
OpenPCDet/pcdet/datasets/kitti/kitti_object_eval_python/kitti_common.py:56
↓ 6 callersFunction_transpose_and_gather_feat
(feat, ind)
OpenPCDet/pcdet/models/model_utils/centernet_utils.py:129
↓ 6 callersFunctioncropPC
(PC, box, offset=0, scale=1.0)
kitty_utils.py:172
↓ 6 callersFunctioncross
OpenPCDet/pcdet/ops/iou3d_nms/src/iou3d_cpu.cpp:59
↓ 6 callersMethodmake_fc_layers
(fc_cfg, input_channels, output_channels)
OpenPCDet/pcdet/models/dense_heads/point_head_template.py:36
↓ 6 callersMethodprepare_data
Args: data_dict: points: optional, (N, 3 + C_in) gt_boxes: optional, (N, 7 + C) [x, y, z, dx, dy,
OpenPCDet/pcdet/datasets/dataset.py:102
↓ 6 callersMethodproposal_layer
Args: batch_dict: batch_size: batch_cls_preds: (B, num_boxes, num_classes | 1) or (N1+N2+..., num
OpenPCDet/pcdet/models/roi_heads/roi_head_template.py:46
↓ 5 callersMethod__init__
Args: code_weights: (#codes) float list if not None. Code-wise weights.
OpenPCDet/pcdet/utils/loss_utils.py:140
↓ 5 callersMethodassign_targets
(self, batch_dict)
OpenPCDet/pcdet/models/roi_heads/roi_head_template.py:104
↓ 5 callersFunctionball_query
pointnet2/_ext-src/src/ball_query.cpp:8
↓ 5 callersMethodgenerate_predicted_boxes
(self, batch_size, pred_dicts)
OpenPCDet/pcdet/models/dense_heads/center_head.py:253
↓ 5 callersMethodget_lidar
(self, idx)
OpenPCDet/pcdet/datasets/kitti/kitti_dataset.py:63
↓ 5 callersMethodlidar_to_rect
:param pts_lidar: (N, 3) :return pts_rect: (N, 3)
OpenPCDet/pcdet/utils/calibration_kitti.py:65
↓ 5 callersFunctionmake_cuda_ext
(name, module, sources)
OpenPCDet/setup.py:17
↓ 5 callersMethodnormalize
(self, wlh)
data_classes.py:123
↓ 5 callersMethodread_val
Read a hyperparameter `key` in the optimizer dictionary.
OpenPCDet/tools/train_utils/optimization/fastai_optim.py:231
↓ 4 callersMethodUR_block_forward
(self, x_lateral, x_bottom, conv_t, conv_m, conv_inv)
OpenPCDet/pcdet/models/backbones_3d/spconv_unet.py:135
↓ 4 callersMethod__init__
Args: radii: list of float, list of radii to group with nsamples: list of int, number of samples in each ball query
OpenPCDet/pcdet/ops/pointnet2/pointnet2_stack/pointnet2_modules.py:32
↓ 4 callersMethod__init__
( self, mlp, radius=None, nsample=None, bn=True, use_xyz=True, sample_method=None )
pointnet2/utils/pointnet2_modules.py:194
↓ 4 callersFunction_gather_feat
(feat, ind, mask=None)
OpenPCDet/pcdet/models/model_utils/centernet_utils.py:118
↓ 4 callersMethodadd_accuracy
(self, val)
metrics.py:125
↓ 4 callersMethodadd_overlap
(self, val)
metrics.py:91
↓ 4 callersMethodassign_targets
Args: gt_boxes: (B, M, 8) range_image_polar: (B, 3, H, W) feature_map_size: (2) [H, W] spatia
OpenPCDet/pcdet/models/dense_heads/center_head.py:159
↓ 4 callersFunctioncfg_from_yaml_file
(cfg_file, config)
OpenPCDet/pcdet/config.py:71
↓ 4 callersMethodcheck_orthogonal
Check that vector (b - a) is orthogonal to the vector (c - a).
OpenPCDet/pcdet/datasets/lyft/lyft_mAP_eval/lyft_eval.py:112
↓ 4 callersMethodcreate
Create an `optim.Optimizer` from `opt_func` with `lr`. Set lr on `layer_groups`.
OpenPCDet/tools/train_utils/optimization/fastai_optim.py:115
↓ 4 callersMethoddecode_torch
Args: box_encodings: (B, N, 7 + C) or (N, 7 + C) [x, y, z, dx, dy, dz, heading or *[cos, sin], ...] anchors: (B, N, 7
OpenPCDet/pcdet/utils/box_coder_utils.py:45
↓ 4 callersMethodencode_torch
Args: boxes: (N, 7 + C) [x, y, z, dx, dy, dz, heading, ...] anchors: (N, 7 + C) [x, y, z, dx, dy, dz, heading or *[co
OpenPCDet/pcdet/utils/box_coder_utils.py:13
↓ 4 callersMethodgenerate_predicted_boxes
Args: batch_size: rois: (B, N, 7) cls_preds: (BN, num_class) box_preds: (BN, code_size)
OpenPCDet/pcdet/models/roi_heads/roi_head_template.py:233
↓ 4 callersMethodgetPCandBBfromIndex
(self, anno_idx)
Dataset.py:211
↓ 4 callersFunctionget_kitti_info_path
(idx, prefix, info_type='image_2', fil
OpenPCDet/pcdet/datasets/kitti/kitti_object_eval_python/kitti_common.py:15
↓ 4 callersMethodget_lr
(self)
OpenPCDet/tools/train_utils/optimization/learning_schedules_fastai.py:86
↓ 4 callersFunctionget_mAP
(prec)
OpenPCDet/pcdet/datasets/kitti/kitti_object_eval_python/eval.py:556
↓ 4 callersFunctionget_mAP_R40
(prec)
OpenPCDet/pcdet/datasets/kitti/kitti_object_eval_python/eval.py:563
↓ 4 callersFunctionget_model
(name, *args, **kwargs)
pointnet2/models/__init__.py:13
↓ 4 callersMethodload_params_from_file
(self, filename, logger, to_cpu=False)
OpenPCDet/pcdet/models/detectors/detector3d_template.py:361
↓ 4 callersFunctionpoints_in_pyramids_mask
(points, pyramids)
OpenPCDet/pcdet/datasets/augmentor/augmentor_utils.py:527
↓ 4 callersMethodrect_to_lidar
:param pts_lidar: (N, 3) :return pts_rect: (N, 3)
OpenPCDet/pcdet/utils/calibration_kitti.py:50
↓ 4 callersMethodset_split
(self, split)
OpenPCDet/pcdet/datasets/kitti/kitti_dataset.py:53
↓ 4 callersFunctiontrangle_area
(a, b, c)
OpenPCDet/pcdet/datasets/kitti/kitti_object_eval_python/rotate_iou.py:18
↓ 4 callersMethodzero_grad
Clear optimizer gradients.
OpenPCDet/tools/train_utils/optimization/fastai_optim.py:154
↓ 3 callersMethod__init__
:param mlp: list of int, spec of the pointnet before the global max_pool :param npoint: int, number of features :param radius
OpenPCDet/pcdet/ops/pointnet2/pointnet2_batch/pointnet2_modules.py:105
↓ 3 callersMethod__init__
(self, code_size=7, encode_angle_by_sincos=False, **kwargs)
OpenPCDet/pcdet/utils/box_coder_utils.py:6
↓ 3 callersMethod__init__
:param mlp: list of int, spec of the pointnet before the global max_pool :param npoint: int, number of features :param radius
pointnet2/_ext-src/pointnet2_3DSSD/pointnet2_modules.py:98
↓ 3 callersMethod__init__
(self, use_xyz=True)
pointnet2/utils/pointnet2_utils.py:449
↓ 3 callersMethod__len__
(self)
OpenPCDet/pcdet/datasets/waymo/waymo_dataset.py:179
↓ 3 callersMethodassign_stack_targets
Args: points: (N1 + N2 + N3 + ..., 4) [bs_idx, x, y, z] gt_boxes: (B, M, 8) extend_gt_boxes: [B, M, 8]
OpenPCDet/pcdet/models/dense_heads/point_head_template.py:49
↓ 3 callersMethodbackward
(ctx, a=None, b=None)
pointnet2/utils/pointnet2_utils.py:140
↓ 3 callersMethodboxes3d_kitti_fakelidar_to_lidar
Args: boxes3d_fakelidar: (N, 7) [x, y, z, w, l, h, r] in old LiDAR coordinates, z is bottom center Returns:
OpenPCDet/pcdet/datasets/waymo/waymo_eval.py:27
↓ 3 callersFunctionbuild_dataloader
(dataset_cfg, class_names, batch_size, dist, root_path=None, workers=4, logger=None, trai
OpenPCDet/pcdet/datasets/__init__.py:47
↓ 3 callersFunctionbuild_network
(model_cfg, num_class, dataset)
OpenPCDet/pcdet/models/__init__.py:16
↓ 3 callersMethodcart_to_hom
:param pts: (N, 3 or 2) :return pts_hom: (N, 4 or 3)
OpenPCDet/pcdet/utils/calibration_kitti.py:42
↓ 3 callersFunctioncheck_numpy_to_torch
(x)
OpenPCDet/tools/visual_utils/visualize_utils.py:13
↓ 3 callersFunctioncheck_numpy_to_torch
(x)
OpenPCDet/pcdet/utils/common_utils.py:15
↓ 3 callersFunctiondraw_corners3d
:param corners3d: (N, 8, 3) :param fig: :param color: :param line_width: :param cls: :param tag: :param max_num: :ret
OpenPCDet/tools/visual_utils/visualize_utils.py:173
↓ 3 callersFunctionestimateOverlap
(box_a, box_b, dim=2)
metrics.py:53
↓ 3 callersFunctioneval_class
Kitti eval. support 2d/bev/3d/aos eval. support 0.5:0.05:0.95 coco AP. Args: gt_annos: dict, must from get_label_annos() in kitti_common.p
OpenPCDet/pcdet/datasets/kitti/kitti_object_eval_python/eval.py:448
↓ 3 callersMethodforward
Args: data_dict: points: (N, 3 + C_in) gt_boxes: optional, (N, 7) [x, y, z, dx, dy, dz, heading]
OpenPCDet/pcdet/datasets/augmentor/data_augmentor.py:228
↓ 3 callersFunctionfunc
(a, x)
kitty_utils.py:266
↓ 3 callersMethodgenerate_recall_record
(box_preds, recall_dict, batch_index, data_dict=None, thresh_list=None)
OpenPCDet/pcdet/models/detectors/detector3d_template.py:287
↓ 3 callersMethodgetAnnotationIndex
(self, index)
Dataset.py:317
↓ 3 callersMethodgetBBandPC
(self, anno)
Dataset.py:43
↓ 3 callersFunctiongetModel
(PCs, boxes, offset=0, scale=1.0, normalize=False)
kitty_utils.py:156
↓ 3 callersMethodget_calib
(self, idx)
OpenPCDet/pcdet/datasets/kitti/kitti_dataset.py:108
↓ 3 callersMethodget_cls_layer_loss
(self)
OpenPCDet/pcdet/models/dense_heads/anchor_head_multi.py:245
↓ 3 callersMethodget_infos
(self, num_workers=4, has_label=True, count_inside_pts=True, sample_id_list=None)
OpenPCDet/pcdet/datasets/kitti/kitti_dataset.py:150
↓ 3 callersMethodget_lidar
(self, sequence_name, sample_idx)
OpenPCDet/pcdet/datasets/waymo/waymo_dataset.py:169
↓ 3 callersFunctionget_pyramids
(boxes)
OpenPCDet/pcdet/datasets/augmentor/augmentor_utils.py:494
↓ 3 callersFunctionget_world_size
()
OpenPCDet/pcdet/utils/commu_utils.py:15
↓ 3 callersFunctiongroup_by_key
(detections, key)
OpenPCDet/pcdet/datasets/lyft/lyft_mAP_eval/lyft_eval.py:214
↓ 3 callersFunctionload_data_to_gpu
(batch_dict)
OpenPCDet/pcdet/models/__init__.py:23
↓ 3 callersFunctionone_hot
(x, num_class=1)
OpenPCDet/pcdet/datasets/augmentor/augmentor_utils.py:519
↓ 3 callersMethodrect_to_img
:param pts_rect: (N, 3) :return pts_img: (N, 2)
OpenPCDet/pcdet/utils/calibration_kitti.py:75
↓ 3 callersMethodreset
(self)
metrics.py:88
next →1–100 of 1,101, ranked by callers