MCPcopy Create free account

hub / github.com/JIA-Lab-research/UVTR / functions

Functions217 in github.com/JIA-Lab-research/UVTR

↓ 1 callersMethodprocess_pts
(self, pts)
projects/mmdet3d_plugin/models/detectors/uvtr_kd_m.py:175
↓ 1 callersMethodremove_points_in_boxes
Remove the points in the sampled bounding boxes. Args: points (:obj:`BasePoints`): Input point cloud array. boxes (np
projects/mmdet3d_plugin/datasets/pipelines/transform_3d.py:611
↓ 1 callersMethodsample_class_v2
Sampling specific categories of bounding boxes. Args: name (str): Class of objects to be sampled. num (int): Number o
projects/mmdet3d_plugin/datasets/pipelines/dbsampler.py:232
↓ 1 callersMethodshow
Results visualization. Args: results (list[dict]): List of bounding boxes results. out_dir (str): Output directory of
projects/mmdet3d_plugin/datasets/nuscenes_dataset.py:650
↓ 1 callersMethodsimple_test
Test function without augmentaiton.
projects/mmdet3d_plugin/models/detectors/uvtr.py:343
↓ 1 callersMethodsimple_test
Test function without augmentaiton.
projects/mmdet3d_plugin/models/detectors/uvtr_kd_cs.py:373
↓ 1 callersMethodsimple_test
Test function without augmentaiton.
projects/mmdet3d_plugin/models/detectors/uvtr_kd_m.py:438
↓ 1 callersMethodsimple_test
Test function without augmentaiton.
projects/mmdet3d_plugin/models/detectors/uvtr_kd_l.py:342
↓ 1 callersMethodsimple_test_pts
Test function of point cloud branch.
projects/mmdet3d_plugin/models/detectors/uvtr.py:332
↓ 1 callersMethodsimple_test_pts
Test function of point cloud branch.
projects/mmdet3d_plugin/models/detectors/uvtr_kd_cs.py:362
↓ 1 callersMethodsimple_test_pts
Test function of point cloud branch.
projects/mmdet3d_plugin/models/detectors/uvtr_kd_m.py:427
↓ 1 callersMethodsimple_test_pts
Test function of point cloud branch.
projects/mmdet3d_plugin/models/detectors/uvtr_kd_l.py:331
↓ 1 callersMethodunified_sample
(self, imgs, lidar2img, points, points_idx, bboxes_3d, sampled_img, sampled_num)
projects/mmdet3d_plugin/datasets/pipelines/transform_3d.py:691
Method__call__
Args: bbox_pred (Tensor): Predicted boxes with normalized coordinates (cx, cy, w, h), which are all in range [0,
projects/mmdet3d_plugin/core/bbox/match_costs/match_cost.py:15
Method__call__
Call function to pad images, masks, semantic segmentation maps. Args: results (dict): Result dict from loading pipeline. R
projects/mmdet3d_plugin/datasets/pipelines/transform_3d.py:49
Method__call__
Call function to normalize images. Args: results (dict): Result dict from loading pipeline. Returns: dict: Nor
projects/mmdet3d_plugin/datasets/pipelines/transform_3d.py:83
Method__call__
Call function to perform photometric distortion on images. Args: results (dict): Result dict from loading pipeline. Return
projects/mmdet3d_plugin/datasets/pipelines/transform_3d.py:133
Method__call__
Call function to pad images, masks, semantic segmentation maps. Args: results (dict): Result dict from loading pipeline. R
projects/mmdet3d_plugin/datasets/pipelines/transform_3d.py:211
Method__call__
(self, results)
projects/mmdet3d_plugin/datasets/pipelines/transform_3d.py:297
Method__call__
Private function to rotate, scale and translate bounding boxes and \ points. Args: input_dict (dict): Result dict from lo
projects/mmdet3d_plugin/datasets/pipelines/transform_3d.py:444
Method__call__
Call function to flip points, values in the ``bbox3d_fields`` and \ also flip 2D image and its annotations. Args: input_d
projects/mmdet3d_plugin/datasets/pipelines/transform_3d.py:539
Method__call__
Call function to sample ground truth objects to the data. Args: input_dict (dict): Result dict from loading pipeline. Re
projects/mmdet3d_plugin/datasets/pipelines/transform_3d.py:625
Method__call__
Call function to augment common fields in results. Args: results (dict): Result dict contains the data to augment. Retur
projects/mmdet3d_plugin/datasets/pipelines/test_time_aug.py:69
Method__call__
Call function to collect keys in results. The keys in ``meta_keys`` will be converted to :obj:`mmcv.DataContainer`. Args:
projects/mmdet3d_plugin/datasets/pipelines/formatting.py:67
Method__call__
Call function to load multi-view image from files. Args: results (dict): Result dict containing multi-view image filenames.
projects/mmdet3d_plugin/datasets/pipelines/loading_3d.py:25
Method__call__
(self, img, label)
projects/mmdet3d_plugin/models/utils/grid_mask.py:20
Method__init__
(self, pc_range, voxel_size=None, post_center_range=None,
projects/mmdet3d_plugin/core/bbox/coders/nms_free_coder.py:21
Method__init__
(self, weight=1.)
projects/mmdet3d_plugin/core/bbox/match_costs/match_cost.py:12
Method__init__
(self, cls_cost=dict(type='ClassificationCost', weight=1.), reg_cost=dict(ty
projects/mmdet3d_plugin/core/bbox/assigners/hungarian_assigner_3d.py:42
Method__init__
(self, ann_file, pipeline=None, data_root=None,
projects/mmdet3d_plugin/datasets/nuscenes_dataset.py:117
Method__init__
(self, mean, std, to_rgb=True)
projects/mmdet3d_plugin/datasets/pipelines/transform_3d.py:78
Method__init__
(self, brightness_delta=32, contrast_range=(0.5, 1.5), satu
projects/mmdet3d_plugin/datasets/pipelines/transform_3d.py:123
Method__init__
(self, scales=[0.5, 1.0, 1.5])
projects/mmdet3d_plugin/datasets/pipelines/transform_3d.py:208
Method__init__
(self, flip_ratio=None, resize_scales=None, crop_sizes=None, training=True)
projects/mmdet3d_plugin/datasets/pipelines/transform_3d.py:245
Method__init__
(self, rot_range=[-0.78539816, 0.78539816], scale_ratio_range=[0.95, 1.05],
projects/mmdet3d_plugin/datasets/pipelines/transform_3d.py:339
Method__init__
(self, flip_ratio_bev_horizontal=0.0, flip_ratio_bev_vertical=0.0,
projects/mmdet3d_plugin/datasets/pipelines/transform_3d.py:499
Method__init__
(self, db_sampler, sample_2d=False, sample_method='depth', modify_points=False)
projects/mmdet3d_plugin/datasets/pipelines/transform_3d.py:601
Method__init__
(self, transforms, img_scale, pts_scale_ratio,
projects/mmdet3d_plugin/datasets/pipelines/test_time_aug.py:33
Method__init__
(self, keys, meta_keys=('filename', 'ori_shape', 'img_shape', 'lidar2img',
projects/mmdet3d_plugin/datasets/pipelines/formatting.py:53
Method__init__
(self, info_path, data_root, rate, prepare
projects/mmdet3d_plugin/datasets/pipelines/dbsampler.py:30
Method__init__
(self, to_float32=False, sweep_num=1, random_sweep=False, color_type='unchanged')
projects/mmdet3d_plugin/datasets/pipelines/loading_3d.py:19
Method__init__
(self, in_channels=[128, 128, 256], out_channels=[256, 256, 256],
projects/mmdet3d_plugin/models/necks/second3d_fpn.py:27
Method__init__
(self, inplace=True)
projects/mmdet3d_plugin/models/backbones/vovnet.py:157
Method__init__
(self, channel, reduction=4)
projects/mmdet3d_plugin/models/backbones/vovnet.py:166
Method__init__
( self, in_ch, stage_ch, concat_ch, layer_per_block, module_name, SE=False, identity=False, depthwise=
projects/mmdet3d_plugin/models/backbones/vovnet.py:181
Method__init__
( self, in_ch, stage_ch, concat_ch, block_per_stage, layer_per_block, stage_num, SE=False, depthwise=F
projects/mmdet3d_plugin/models/backbones/vovnet.py:234
Method__init__
(self, in_channels=128, out_channels=[128, 128, 256], layer
projects/mmdet3d_plugin/models/backbones/second_3d.py:24
Method__init__
(self, *args, unified_conv=None, view_cfg=None,
projects/mmdet3d_plugin/models/dense_heads/uvtr_head.py:30
Method__init__
(self, *args, view_cfg=None, with_box_refine=False,
projects/mmdet3d_plugin/models/dense_heads/uvtr_kd_head.py:31
Method__init__
(self, *args, return_intermediate=False, **kwargs)
projects/mmdet3d_plugin/models/utils/uni3d_detr.py:97
Method__init__
(self, embed_dims=256, num_heads=8, num_points=1,
projects/mmdet3d_plugin/models/utils/uni3d_detr.py:162
Method__init__
(self, use_h, use_w, rotate = 1, offset=False, ratio = 0.5, mode=0, prob = 1.)
projects/mmdet3d_plugin/models/utils/grid_mask.py:70
Method__init__
(self, embed_dims=256, num_levels=4, num_points=5,
projects/mmdet3d_plugin/models/utils/uni3d_viewtrans.py:179
Method__init__
(self, use_grid_mask=False, pts_voxel_layer=None, pts_voxel
projects/mmdet3d_plugin/models/detectors/uvtr.py:19
Method__init__
(self, distill_type='Sweep_2D', use_grid_mask=False, teache
projects/mmdet3d_plugin/models/detectors/uvtr_kd_cs.py:21
Method__init__
(self, distill_type='Multi_2D', use_grid_mask=False, pts_vo
projects/mmdet3d_plugin/models/detectors/uvtr_kd_m.py:21
Method__init__
(self, distill_type='3D_2D', use_grid_mask=False, pts_voxel
projects/mmdet3d_plugin/models/detectors/uvtr_kd_l.py:19
Method__init__
(self, in_channels, sparse_shape, order=('conv', 'norm', 'a
projects/mmdet3d_plugin/models/pts_encoder/sparse_encoder_hd.py:32
Method__repr__
(self)
projects/mmdet3d_plugin/datasets/pipelines/transform_3d.py:59
Method__repr__
(self)
projects/mmdet3d_plugin/datasets/pipelines/transform_3d.py:97
Method__repr__
(self)
projects/mmdet3d_plugin/datasets/pipelines/transform_3d.py:192
Method__repr__
(self)
projects/mmdet3d_plugin/datasets/pipelines/transform_3d.py:237
Method__repr__
str: Return a string that describes the module.
projects/mmdet3d_plugin/datasets/pipelines/transform_3d.py:474
Method__repr__
str: Return a string that describes the module.
projects/mmdet3d_plugin/datasets/pipelines/transform_3d.py:582
Method__repr__
str: Return a string that describes the module.
projects/mmdet3d_plugin/datasets/pipelines/transform_3d.py:775
Method__repr__
str: Return a string that describes the module.
projects/mmdet3d_plugin/datasets/pipelines/test_time_aug.py:118
Method__repr__
str: Return a string that describes the module.
projects/mmdet3d_plugin/datasets/pipelines/formatting.py:91
Method__repr__
str: Return a string that describes the module.
projects/mmdet3d_plugin/datasets/pipelines/loading_3d.py:106
Method_build_default_pipeline
Build the default pipeline for this dataset.
projects/mmdet3d_plugin/datasets/nuscenes_dataset.py:629
Method_get_clones
(module, N)
projects/mmdet3d_plugin/models/dense_heads/uvtr_head.py:100
Method_get_clones
(module, N)
projects/mmdet3d_plugin/models/dense_heads/uvtr_kd_head.py:94
Method_get_target_single
Compute regression and classification targets for one image. Outputs from a single decoder layer of a single feature level are used. A
projects/mmdet3d_plugin/models/dense_heads/uvtr_head.py:223
Method_get_target_single
Compute regression and classification targets for one image. Outputs from a single decoder layer of a single feature level are used. A
projects/mmdet3d_plugin/models/dense_heads/uvtr_kd_head.py:238
Method_init_layers
Initialize classification branch and regression branch of head.
projects/mmdet3d_plugin/models/dense_heads/uvtr_head.py:83
Method_init_layers
Initialize classification branch and regression branch of head.
projects/mmdet3d_plugin/models/dense_heads/uvtr_kd_head.py:77
Method_initialize_weights
(self)
projects/mmdet3d_plugin/models/backbones/vovnet.py:336
Functioncreate_nuscenes_infos
Create info file of nuscene dataset. Given the raw data, generate its related info file in pkl format. Args: root_path (str): Path o
extra_tools/data_converter/nuscenes_converter.py:30
Methodencode
(self)
projects/mmdet3d_plugin/core/bbox/coders/nms_free_coder.py:36
Functionexport_2d_annotation
Export 2d annotation from the info file and raw data. Args: root_path (str): Root path of the raw data. info_path (str): Path of
extra_tools/data_converter/nuscenes_converter.py:368
Methodfilter_by_difficulty
Filter ground truths by difficulties. Args: db_infos (dict): Info of groundtruth database. removed_difficulty (list):
projects/mmdet3d_plugin/datasets/pipelines/dbsampler.py:86
Methodfilter_by_min_points
Filter ground truths by number of points in the bbox. Args: db_infos (dict): Info of groundtruth database. min_gt_poi
projects/mmdet3d_plugin/datasets/pipelines/dbsampler.py:105
Methodforward
Forward function. Args: x (torch.Tensor): 4D Tensor in (N, C, H, W) shape. Returns: list[torch.Tensor]: Mult
projects/mmdet3d_plugin/models/necks/second3d_fpn.py:113
Methodforward
(self, x)
projects/mmdet3d_plugin/models/backbones/vovnet.py:161
Methodforward
(self, x)
projects/mmdet3d_plugin/models/backbones/vovnet.py:172
Methodforward
(self, x)
projects/mmdet3d_plugin/models/backbones/vovnet.py:210
Methodforward
(self, x)
projects/mmdet3d_plugin/models/backbones/vovnet.py:341
Methodforward
Forward function. Args: x (torch.Tensor): Input with shape (N, C, H, W). Returns: tuple[torch.Tensor]: Multi
projects/mmdet3d_plugin/models/backbones/second_3d.py:89
Methodforward
Forward function. Args: mlvl_feats (tuple[Tensor]): Features from the upstream network, each is a 5D-tensor with s
projects/mmdet3d_plugin/models/dense_heads/uvtr_head.py:130
Methodforward
Forward function. Args: mlvl_feats (tuple[Tensor]): Features from the upstream network, each is a 5D-tensor with s
projects/mmdet3d_plugin/models/dense_heads/uvtr_kd_head.py:133
Methodforward
(self, pts_value, img_value, query_embed, reg_
projects/mmdet3d_plugin/models/utils/uni3d_detr.py:53
Methodforward
Forward function for `UniTransformerDecoder`. Args: query (Tensor): Input query with shape `(num_query, b
projects/mmdet3d_plugin/models/utils/uni3d_detr.py:101
Methodforward
Forward Function of UniCrossAtten. Args: query (Tensor): Query of Transformer with shape (num_query, bs, embed_dim
projects/mmdet3d_plugin/models/utils/uni3d_detr.py:212
Methodforward
(self, x)
projects/mmdet3d_plugin/models/utils/grid_mask.py:84
Methodforward
Forward function for `Uni3DViewTrans`. Args: mlvl_feats (list(Tensor)): Input queries from different level. Each e
projects/mmdet3d_plugin/models/utils/uni3d_viewtrans.py:114
Methodforward
(self, mlvl_feats, img_depth=None, **kwargs)
projects/mmdet3d_plugin/models/utils/uni3d_viewtrans.py:206
Methodforward
Calls either forward_train or forward_test depending on whether return_loss=True. Note this setting will change the expected inputs. W
projects/mmdet3d_plugin/models/detectors/uvtr.py:244
Methodforward
Calls either forward_train or forward_test depending on whether return_loss=True. Note this setting will change the expected inputs. W
projects/mmdet3d_plugin/models/detectors/uvtr_kd_cs.py:274
Methodforward
Calls either forward_train or forward_test depending on whether return_loss=True. Note this setting will change the expected inputs. W
projects/mmdet3d_plugin/models/detectors/uvtr_kd_m.py:339
Methodforward
Calls either forward_train or forward_test depending on whether return_loss=True. Note this setting will change the expected inputs. W
projects/mmdet3d_plugin/models/detectors/uvtr_kd_l.py:243
Methodforward
Forward of SparseEncoder. Args: voxel_features (torch.float32): Voxel features in shape (N, C). coors (torch.int32):
projects/mmdet3d_plugin/models/pts_encoder/sparse_encoder_hd.py:103
← previousnext →101–200 of 217, ranked by callers