MCPcopy Create free account

hub / github.com/Jasonkks/PTTR / functions

Functions1,101 in github.com/Jasonkks/PTTR

Method__init__
Initializes DDNDeepLabV3 model Args: backbone_name: string, ResNet Backbone Name [ResNet50/ResNet101]
OpenPCDet/pcdet/models/backbones_3d/vfe/image_vfe_modules/ffn/ddn/ddn_deeplabv3.py:11
Method__init__
(self, model_cfg, input_channels)
OpenPCDet/pcdet/models/backbones_2d/base_bev_backbone.py:7
Method__init__
Initializes 2D convolution collapse module Args: model_cfg: EasyDict, Model configuration grid_size: (X, Y, Z
OpenPCDet/pcdet/models/backbones_2d/map_to_bev/conv2d_collapse.py:9
Method__init__
(self, model_cfg, grid_size, **kwargs)
OpenPCDet/pcdet/models/backbones_2d/map_to_bev/pointpillar_scatter.py:6
Method__init__
(self, model_cfg, **kwargs)
OpenPCDet/pcdet/models/backbones_2d/map_to_bev/height_compression.py:5
Method__init__
( self, args: List[int], *, bn: bool = False, acti
pointnet2/_ext-src/pointnet2_3DSSD/pytorch_utils.py:7
Method__init__
( self, in_size, out_size, kernel_size, stride,
pointnet2/_ext-src/pointnet2_3DSSD/pytorch_utils.py:37
Method__init__
(self, in_size, batch_norm=None, name="")
pointnet2/_ext-src/pointnet2_3DSSD/pytorch_utils.py:106
Method__init__
(self, in_size: int, *, name: str = "")
pointnet2/_ext-src/pointnet2_3DSSD/pytorch_utils.py:116
Method__init__
(self, in_size: int, name: str = "")
pointnet2/_ext-src/pointnet2_3DSSD/pytorch_utils.py:122
Method__init__
( self, in_size: int, out_size: int, *, kernel_siz
pointnet2/_ext-src/pointnet2_3DSSD/pytorch_utils.py:128
Method__init__
( self, in_size: int, out_size: int, *, kernel_siz
pointnet2/_ext-src/pointnet2_3DSSD/pytorch_utils.py:165
Method__init__
:param radius: float, radius of ball :param nsample: int, maximum number of features to gather in the ball :param use_xyz:
pointnet2/_ext-src/pointnet2_3DSSD/pointnet2_utils.py:293
Method__init__
(self)
pointnet2/_ext-src/pointnet2_3DSSD/pointnet2_modules.py:12
Method__init__
:param npoint: int :param radii: list of float, list of radii to group with :param nsamples: list of int, number of samples i
pointnet2/_ext-src/pointnet2_3DSSD/pointnet2_modules.py:61
Method__init__
:param mlp: list of int :param bn: whether to use batchnorm
pointnet2/_ext-src/pointnet2_3DSSD/pointnet2_modules.py:119
Method__init__
(self, p=0.5, inplace=False)
pointnet2/utils/pointnet2_utils.py:35
Method__init__
(self, radius, nsample, use_xyz=True)
pointnet2/utils/pointnet2_utils.py:337
Method__init__
(self, radius, nsample, use_xyz=True)
pointnet2/utils/pointnet2_utils.py:393
Method__init__
(self, sample_method=None)
pointnet2/utils/pointnet2_modules.py:49
Method__init__
(self, radii, nsamples, mlps, bn=True, use_xyz=True, vote=False, sample_method=None)
pointnet2/utils/pointnet2_modules.py:152
Method__init__
( self, mlp, radius=None, nsample=None, bn=True, use_xyz=True, vote=True, sample_method=None
pointnet2/utils/pointnet2_modules.py:226
Method__init__
(self, mlp, bn=True)
pointnet2/utils/pointnet2_modules.py:314
Method__init__
(self, input_channel=3, num_pos_feats=256)
pointnet2/models/pointnet_transformer.py:104
Method__init__
(self, input_channels=3, use_xyz=True, objective = False, i
pointnet2/models/pointnet_transformer.py:121
Method__init__
(self, size_average=True, eps=1e-5, scale=1.0)
pointnet2/models/transformer.py:15
Method__init__
(self, d_model=512, nhead=1, activation="relu")
pointnet2/models/transformer.py:66
Method__init__
(self, multihead_attn, FFN, d_model, self_posembed=None)
pointnet2/models/transformer.py:99
Method__init__
(self, multihead_attn, FFN, d_model=512, num_encoder_layers=6,
pointnet2/models/transformer.py:132
Method__init__
(self, multihead_attn, FFN, d_model, self_posembed=None)
pointnet2/models/transformer.py:156
Method__init__
(self, multihead_attn, FFN, d_model=512, num_decoder_layers=6,
pointnet2/models/transformer.py:205
Method__init__
(self, d_model, dim_feedforward, dropout=0.1)
pointnet2/models/multihead_attention.py:9
Method__init__
(self, feature_dim=512, n_head=8, key_feature_dim=64, extra_nonlinear=True)
pointnet2/models/multihead_attention.py:28
Method__iter__
(self)
OpenPCDet/pcdet/datasets/__init__.py:30
Method__len__
(self)
sequence_dataset.py:81
Method__len__
(self)
Dataset.py:356
Method__len__
(self)
OpenPCDet/pcdet/datasets/nuscenes/nuscenes_dataset.py:111
Method__repr__
(self)
data_classes.py:167
Method__repr__
(self)
OpenPCDet/tools/train_utils/optimization/fastai_optim.py:131
Method__repr__
(self)
OpenPCDet/pcdet/datasets/lyft/lyft_mAP_eval/lyft_eval.py:197
Method__setstate__
(self, d)
OpenPCDet/pcdet/datasets/dataset.py:57
Method__setstate__
(self, d)
OpenPCDet/pcdet/datasets/augmentor/data_augmentor.py:40
Method__setstate__
(self, d)
OpenPCDet/pcdet/datasets/augmentor/database_sampler.py:56
Function_get_activation_fn
Return an activation function given a string
pointnet2/models/transformer.py:230
Method_random_sample_pts
(pc, num)
waymo_dataset.py:167
Method_random_sample_pts
(pc, num)
Dataset.py:261
Function_transform
(self, pc, T)
test_tracking_waymo.py:28
Methodabsolute_coordinates_encoding
(self, points=None)
OpenPCDet/pcdet/datasets/processor/point_feature_encoder.py:43
MethodaddData
(self, data, score)
searchspace.py:14
MethodaddData
(self, data, score)
searchspace.py:71
MethodaddData
(self, data, score)
searchspace.py:93
MethodaddData
(self, data, score)
searchspace.py:157
Functionall_reduce
(data, op="sum", average=False)
OpenPCDet/pcdet/utils/commu_utils.py:148
Functionannealing_cos
Cosine anneal from `start` to `end` as pct goes from 0.0 to 1.0.
OpenPCDet/tools/train_utils/optimization/learning_schedules_fastai.py:53
Methodassign_targets
Args: gt_boxes: (B, M, 8) Returns:
OpenPCDet/pcdet/models/dense_heads/anchor_head_template.py:89
Methodassign_targets
Args: all_anchors: [(N, 7), ...] gt_boxes: (B, M, 8) Returns:
OpenPCDet/pcdet/models/dense_heads/target_assigner/axis_aligned_target_assigner.py:36
Methodassign_targets
Args: anchors: [(N, 7), ...] gt_boxes: (B, M, 8) Returns:
OpenPCDet/pcdet/models/dense_heads/target_assigner/atss_target_assigner.py:16
Methodaverage
(self)
metrics.py:142
Functionaverage_reduce_value
(data)
OpenPCDet/pcdet/utils/commu_utils.py:143
Methodbackward
(xyz, a=None)
OpenPCDet/pcdet/ops/pointnet2/pointnet2_batch/pointnet2_utils.py:32
Methodbackward
(ctx, grad_out)
OpenPCDet/pcdet/ops/pointnet2/pointnet2_batch/pointnet2_utils.py:63
Methodbackward
:param ctx: :param grad_out: (B, C, N) tensor with gradients of outputs :return: grad_features: (B, C, M) tensor
OpenPCDet/pcdet/ops/pointnet2/pointnet2_batch/pointnet2_utils.py:134
Methodbackward
:param ctx: :param grad_out: (B, C, npoint, nsample) tensor of the gradients of the output from forward :return:
OpenPCDet/pcdet/ops/pointnet2/pointnet2_batch/pointnet2_utils.py:180
Methodbackward
(ctx, a=None)
OpenPCDet/pcdet/ops/pointnet2/pointnet2_batch/pointnet2_utils.py:224
Methodbackward
(ctx, a=None)
OpenPCDet/pcdet/ops/pointnet2/pointnet2_stack/voxel_query_utils.py:45
Methodbackward
(ctx, a=None)
OpenPCDet/pcdet/ops/pointnet2/pointnet2_stack/pointnet2_utils.py:41
Methodbackward
Args: ctx: grad_out: (M1 + M2 ..., C, nsample) tensor of the gradients of the output from forward Returns:
OpenPCDet/pcdet/ops/pointnet2/pointnet2_stack/pointnet2_utils.py:85
Methodbackward
(xyz, a=None)
OpenPCDet/pcdet/ops/pointnet2/pointnet2_stack/pointnet2_utils.py:180
Methodbackward
(xyz, a=None)
OpenPCDet/pcdet/ops/pointnet2/pointnet2_stack/pointnet2_utils.py:217
Methodbackward
(ctx, a=None, b=None)
OpenPCDet/pcdet/ops/pointnet2/pointnet2_stack/pointnet2_utils.py:253
Methodbackward
Args: ctx: grad_out: (N1 + N2 ..., C) Returns: grad_features: (M1 + M2 ..., C)
OpenPCDet/pcdet/ops/pointnet2/pointnet2_stack/pointnet2_utils.py:282
Methodbackward
Args: ctx: grad_new_features: (M1 + M2 ..., num_c_out), num_c_out = num_c_out_each_grid * num_total_grids Re
OpenPCDet/pcdet/ops/pointnet2/pointnet2_stack/pointnet2_utils.py:428
Methodbackward
(ctx, grad_out)
OpenPCDet/pcdet/ops/roipoint_pool3d/roipoint_pool3d_utils.py:62
Methodbackward
:param grad_out: (N, out_x, out_y, out_z, C) :return: grad_in: (npoints, C)
OpenPCDet/pcdet/ops/roiaware_pool3d/roiaware_pool3d_utils.py:96
Methodbackward
(xyz, a=None)
pointnet2/_ext-src/pointnet2_3DSSD/pointnet2_utils.py:32
Methodbackward
(xyz, a=None)
pointnet2/_ext-src/pointnet2_3DSSD/pointnet2_utils.py:61
Methodbackward
(ctx, grad_out)
pointnet2/_ext-src/pointnet2_3DSSD/pointnet2_utils.py:92
Methodbackward
(ctx, a=None, b=None)
pointnet2/_ext-src/pointnet2_3DSSD/pointnet2_utils.py:130
Methodbackward
:param ctx: :param grad_out: (B, C, N) tensor with gradients of outputs :return: grad_features: (B, C, M) tensor
pointnet2/_ext-src/pointnet2_3DSSD/pointnet2_utils.py:163
Methodbackward
:param ctx: :param grad_out: (B, C, npoint, nsample) tensor of the gradients of the output from forward :return:
pointnet2/_ext-src/pointnet2_3DSSD/pointnet2_utils.py:209
Methodbackward
(ctx, a=None)
pointnet2/_ext-src/pointnet2_3DSSD/pointnet2_utils.py:253
Methodbackward
(ctx, a=None)
pointnet2/_ext-src/pointnet2_3DSSD/pointnet2_utils.py:285
Methodbackward
(xyz, a=None)
pointnet2/utils/pointnet2_utils.py:71
Methodbackward
(ctx, grad_out)
pointnet2/utils/pointnet2_utils.py:104
Methodbackward
r""" Parameters ---------- grad_out : torch.Tensor (B, c, n) tensor with gradients of ouputs Returns
pointnet2/utils/pointnet2_utils.py:176
Methodbackward
r""" Parameters ---------- grad_out : torch.Tensor (B, C, npoint, nsample) tensor of the gradients of the output
pointnet2/utils/pointnet2_utils.py:232
Methodbackward
(ctx, a=None)
pointnet2/utils/pointnet2_utils.py:284
Methodbackward
(ctx, a=None)
pointnet2/utils/pointnet2_utils.py:318
Functionball_query_dilated_wrapper_fast
pointnet2/_ext-src/pointnet2_3DSSD/src/ball_query.cpp:28
Functionball_query_wrapper_fast
OpenPCDet/pcdet/ops/pointnet2/pointnet2_batch/src/ball_query.cpp:32
Functionball_query_wrapper_fast
pointnet2/_ext-src/pointnet2_3DSSD/src/ball_query.cpp:14
Functionball_query_wrapper_stack
OpenPCDet/pcdet/ops/pointnet2/pointnet2_stack/src/ball_query.cpp:31
Methodbeta
(self)
OpenPCDet/tools/train_utils/optimization/fastai_optim.py:190
Functionbin_depths
Converts depth map into bin indices Args: depth_map: (H, W), Depth Map mode: string, Discretiziation mode (See https://arxiv.
OpenPCDet/pcdet/utils/transform_utils.py:56
Methodbottom_corners
Returns the four bottom corners. :return: <np.float: 3, 4>. Bottom corners. First two face forward, last two face backwards.
data_classes.py:253
Functionboxes3d_kitti_camera_to_imageboxes
:param boxes3d: (N, 7) [x, y, z, l, h, w, r] in rect camera coords :param calib: :return: box_2d_preds: (N, 4) [x1, y1, x2, y2]
OpenPCDet/pcdet/utils/box_utils.py:226
Functionboxes3d_kitti_camera_to_lidar
Args: boxes3d_camera: (N, 7) [x, y, z, l, h, w, r] in rect camera coords calib: Returns: boxes3d_lidar: [x, y, z, dx
OpenPCDet/pcdet/utils/box_utils.py:92
Functionboxes3d_kitti_lidar_to_fakelidar
Args: boxes3d_lidar: (N, 7) [x, y, z, dx, dy, dz, heading], (x, y, z) is the box center Returns: boxes3d_fakelidar: [x, y, z
OpenPCDet/pcdet/utils/box_utils.py:128
Functionboxes3d_lidar_to_kitti_camera
:param boxes3d_lidar: (N, 7) [x, y, z, dx, dy, dz, heading], (x, y, z) is the box center :param calib: :return: boxes3d_camera: (
OpenPCDet/pcdet/utils/box_utils.py:161
Functionboxes3d_nearest_bev_iou
Args: boxes_a: (N, 7) [x, y, z, dx, dy, dz, heading] boxes_b: (N, 7) [x, y, z, dx, dy, dz, heading] Returns:
OpenPCDet/pcdet/utils/box_utils.py:286
← previousnext →601–700 of 1,101, ranked by callers