MCPcopy Create free account

hub / github.com/WangYueFt/detr3d / functions

Functions250 in github.com/WangYueFt/detr3d

↓ 1 callersMethodinit_weights
Initialize weights of the DeformDETR head.
projects/mmdet3d_plugin/models/dense_heads/detr3d_head.py:96
↓ 1 callersFunctionkitti_data_prep
Prepare data related to Kitti dataset. Related data consists of '.pkl' files recording basic infos, 2D annotations and groundtruth database.
tools/create_data.py:12
↓ 1 callersFunctionkitti_result_line
(result_dict, precision=4)
tools/data_converter/kitti_data_utils.py:512
↓ 1 callersFunctionload_json_logs
(json_logs)
tools/analysis_tools/analyze_logs.py:168
↓ 1 callersMethodloss_single
Loss function for outputs from a single decoder layer of a single feature level. Args: cls_scores (Tensor): Box score logi
projects/mmdet3d_plugin/models/dense_heads/detr3d_head.py:272
↓ 1 callersMethodloss_single
Loss function for outputs from a single decoder layer of a single feature level. Args: cls_scores (Tensor): Box score logi
projects/mmdet3d_plugin/models/dense_heads/dgcnn3d_head.py:309
↓ 1 callersFunctionmain
()
tools/train.py:94
↓ 1 callersFunctionmain
()
tools/test.py:102
↓ 1 callersFunctionmain
Convert keys in checkpoints for VoteNet. There can be some breaking changes during the development of mmdetection3d, and this tool is used fo
tools/model_converters/convert_votenet_checkpoints.py:70
↓ 1 callersFunctionmain
()
tools/model_converters/regnet2mmdet.py:80
↓ 1 callersFunctionmain
()
tools/model_converters/publish_model.py:29
↓ 1 callersFunctionmain
()
tools/analysis_tools/benchmark.py:30
↓ 1 callersFunctionmain
()
tools/analysis_tools/analyze_logs.py:188
↓ 1 callersFunctionmain
()
tools/analysis_tools/get_flops.py:43
↓ 1 callersFunctionmain
()
tools/misc/browse_dataset.py:184
↓ 1 callersFunctionmain
()
tools/misc/fuse_conv_bn.py:57
↓ 1 callersFunctionmain
()
tools/misc/visualize_results.py:21
↓ 1 callersFunctionmain
()
tools/misc/print_config.py:16
↓ 1 callersFunctionmain
()
tools/data_converter/nuimage_converter.py:215
↓ 1 callersFunctionparse_args
()
tools/train.py:25
↓ 1 callersFunctionparse_args
()
tools/test.py:20
↓ 1 callersFunctionparse_args
()
tools/model_converters/convert_votenet_checkpoints.py:11
↓ 1 callersFunctionparse_args
()
tools/model_converters/publish_model.py:7
↓ 1 callersFunctionparse_args
()
tools/analysis_tools/benchmark.py:14
↓ 1 callersFunctionparse_args
()
tools/analysis_tools/analyze_logs.py:158
↓ 1 callersFunctionparse_args
()
tools/analysis_tools/get_flops.py:14
↓ 1 callersFunctionparse_args
()
tools/misc/browse_dataset.py:15
↓ 1 callersFunctionparse_args
()
tools/misc/fuse_conv_bn.py:47
↓ 1 callersFunctionparse_args
()
tools/misc/visualize_results.py:9
↓ 1 callersFunctionparse_args
()
tools/misc/print_config.py:6
↓ 1 callersFunctionparse_args
()
tools/data_converter/nuimage_converter.py:32
↓ 1 callersFunctionparse_config
Parse config from strings. Args: config_strings (string): strings of model config. Returns: Config: model config
tools/model_converters/convert_votenet_checkpoints.py:20
↓ 1 callersFunctionprocess_checkpoint
(in_file, out_file)
tools/model_converters/publish_model.py:16
↓ 1 callersFunctionprocess_img_anno
(img_info)
tools/data_converter/nuimage_converter.py:178
↓ 1 callersFunctionrandom_sampling
Random sampling. Sampling point cloud to a certain number of points. Args: points (ndarray): Point cloud. num_points (int):
tools/data_converter/sunrgbd_data_utils.py:9
↓ 1 callersFunctions3dis_data_prep
Prepare the info file for s3dis dataset. Args: root_path (str): Path of dataset root. info_prefix (str): The prefix of info filen
tools/create_data.py:116
↓ 1 callersMethodsave_calib
Parse and save the calibration data. Args: frame (:obj:`Frame`): Open dataset frame proto. file_idx (int): Current fi
tools/data_converter/waymo_converter.py:144
↓ 1 callersMethodsave_image
Parse and save the images in png format. Args: frame (:obj:`Frame`): Open dataset frame proto. file_idx (int): Curren
tools/data_converter/waymo_converter.py:129
↓ 1 callersMethodsave_label
Parse and save the label data in txt format. The relation between waymo and kitti coordinates is noteworthy: 1. x, y, z correspond to
tools/data_converter/waymo_converter.py:250
↓ 1 callersMethodsave_lidar
Parse and save the lidar data in psd format. Args: frame (:obj:`Frame`): Open dataset frame proto. file_idx (int): Cu
tools/data_converter/waymo_converter.py:200
↓ 1 callersMethodsave_pose
Parse and save the pose data. Note that SDC's own pose is not included in the regular training of KITTI dataset. KITTI raw dataset co
tools/data_converter/waymo_converter.py:350
↓ 1 callersFunctionscannet_data_prep
Prepare the info file for scannet dataset. Args: root_path (str): Path of dataset root. info_prefix (str): The prefix of info fil
tools/create_data.py:103
↓ 1 callersFunctionshow_det_data
Visualize 3D point cloud and 3D bboxes.
tools/misc/browse_dataset.py:90
↓ 1 callersFunctionshow_proj_bbox_img
Visualize 3D bboxes on 2D image by projection.
tools/misc/browse_dataset.py:124
↓ 1 callersFunctionshow_seg_data
Visualize 3D point cloud and segmentation mask.
tools/misc/browse_dataset.py:107
↓ 1 callersMethodsimple_test
Test function without augmentaiton.
projects/mmdet3d_plugin/models/detectors/detr3d.py:181
↓ 1 callersMethodsimple_test_pts
Test function of point cloud branch.
projects/mmdet3d_plugin/models/detectors/detr3d.py:170
↓ 1 callersFunctionsunrgbd_data_prep
Prepare the info file for sunrgbd dataset. Args: root_path (str): Path of dataset root. info_prefix (str): The prefix of info fil
tools/create_data.py:129
↓ 1 callersFunctionto_depth_mode
Convert points and bboxes to Depth Coord and Depth Box mode.
tools/misc/browse_dataset.py:79
↓ 1 callersFunctionwaymo_data_prep
Prepare the info file for waymo dataset. Args: root_path (str): Path of dataset root. info_prefix (str): The prefix of info filen
tools/create_data.py:142
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:41
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:75
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:125
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:205
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:233
Method__call__
(self, results)
projects/mmdet3d_plugin/datasets/pipelines/transform_3d.py:267
Method__call__
(self, img, label)
projects/mmdet3d_plugin/models/utils/grid_mask.py:20
Method__init__
(self, line)
tools/data_converter/sunrgbd_data_utils.py:35
Method__init__
(self, root_path, split='train', use_v1=False)
tools/data_converter/sunrgbd_data_utils.py:70
Method__init__
(self, root_path, split='Area_1')
tools/data_converter/s3dis_data_utils.py:19
Method__init__
(self, data_root, ann_file, split='Area_1',
tools/data_converter/s3dis_data_utils.py:163
Method__init__
(self, load_dir, save_dir, prefix, workers
tools/data_converter/waymo_converter.py:38
Method__init__
(self, root_path, split='train')
tools/data_converter/scannet_data_utils.py:19
Method__init__
(self, data_root, ann_file, split='train',
tools/data_converter/scannet_data_utils.py:210
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, size=None, size_divisor=None, pad_val=0)
projects/mmdet3d_plugin/datasets/pipelines/transform_3d.py:19
Method__init__
(self, mean, std, to_rgb=True)
projects/mmdet3d_plugin/datasets/pipelines/transform_3d.py:70
Method__init__
(self, brightness_delta=32, contrast_range=(0.5, 1.5), satu
projects/mmdet3d_plugin/datasets/pipelines/transform_3d.py:115
Method__init__
(self, size=None)
projects/mmdet3d_plugin/datasets/pipelines/transform_3d.py:202
Method__init__
(self, scales=[0.5, 1.0, 1.5])
projects/mmdet3d_plugin/datasets/pipelines/transform_3d.py:230
Method__init__
(self, flip_ratio=0.5)
projects/mmdet3d_plugin/datasets/pipelines/transform_3d.py:264
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, *args, with_box_refine=False, as_two_stage=False,
projects/mmdet3d_plugin/models/dense_heads/detr3d_head.py:28
Method__init__
(self, *args, with_box_refine=False, as_two_stage=False,
projects/mmdet3d_plugin/models/dense_heads/dgcnn3d_head.py:28
Method__init__
(self, num_feature_levels=4, num_cams=6, two_stage_num_prop
projects/mmdet3d_plugin/models/utils/detr3d_transformer.py:47
Method__init__
(self, *args, return_intermediate=False, **kwargs)
projects/mmdet3d_plugin/models/utils/detr3d_transformer.py:151
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, num_heads, dropout=0.,
projects/mmdet3d_plugin/models/utils/dgcnn_attn.py:21
Method__init__
(self, *args, return_intermediate=False, **kwargs)
projects/mmdet3d_plugin/models/utils/detr.py:33
Method__init__
(self, pts_voxel_layer=None, pts_voxel_encoder=None, pts_mi
projects/mmdet3d_plugin/models/detectors/obj_dgcnn.py:12
Method__init__
(self, use_grid_mask=False, pts_voxel_layer=None, pts_voxel
projects/mmdet3d_plugin/models/detectors/detr3d.py:14
Method__len__
(self)
tools/data_converter/sunrgbd_data_utils.py:95
Method__len__
(self)
tools/data_converter/s3dis_data_utils.py:42
Method__len__
Length of the filename list.
tools/data_converter/waymo_converter.py:125
Method__len__
(self)
tools/data_converter/scannet_data_utils.py:44
Method__repr__
(self)
projects/mmdet3d_plugin/datasets/pipelines/transform_3d.py:51
Method__repr__
(self)
projects/mmdet3d_plugin/datasets/pipelines/transform_3d.py:89
Method__repr__
(self)
projects/mmdet3d_plugin/datasets/pipelines/transform_3d.py:184
Method__repr__
(self)
projects/mmdet3d_plugin/datasets/pipelines/transform_3d.py:217
Method__repr__
(self)
projects/mmdet3d_plugin/datasets/pipelines/transform_3d.py:255
Method_get_clones
(module, N)
projects/mmdet3d_plugin/models/dense_heads/detr3d_head.py:75
Method_get_clones
(module, N)
projects/mmdet3d_plugin/models/dense_heads/dgcnn3d_head.py:74
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/detr3d_head.py:165
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/dgcnn3d_head.py:202
← previousnext →101–200 of 250, ranked by callers