MCPcopy Create free account

hub / github.com/WHU-USI3DV/MENet / functions

Functions243 in github.com/WHU-USI3DV/MENet

↓ 9 callersFunctionfreeze_module
(module)
menet/utils/weights_utils.py:10
↓ 8 callersFunctionload_module_from
(model, ckpt_path, prefix, extra_prefix=None, logger=None, map_location='cpu')
menet/utils/weights_utils.py:15
↓ 6 callersMethod__init__
(self, embed_dims, num_heads, window_size,
menet/models/backbone/swin.py:260
↓ 4 callersMethod__init__
(self, in_channels, depthwise_spatial=False, sync_bn=False)
menet/models/attentions/bam.py:106
↓ 4 callersFunctionbuild_model
A function warpper for building 3D detector or segmentor according to cfg. Should be deprecated in the future.
menet/models/builder.py:61
↓ 4 callersFunctiongather_map
(data)
menet/models/encoder/map_encoder.py:8
↓ 4 callersMethodget_bboxes
Generate bboxes from bbox head predictions. Args: preds_dicts (tuple[list[dict]]): Prediction results. img_metas (lis
menet/models/head/centerpoint.py:23
↓ 4 callersFunctionobtain_sensor2top
Obtain the info with RT matric from general sensor to Top LiDAR. Args: nusc (class): Dataset class in the nuScenes dataset. senso
tools/data_converter/nuscenes_converter.py:296
↓ 3 callersMethod__init__
( self, spatial_transformer=dict( conv_group_cfg = dict(
menet/models/encoder/map_encoder.py:104
↓ 3 callersFunctionbuild_conv_group
(cfg)
menet/models/builder.py:58
↓ 3 callersMethodextract_feature
input_data (dict): sensors' data - "imgs" - "map_mask"
menet/models/fusion_models/bevfusion.py:61
↓ 3 callersMethodget_data_info
Get data info according to the given index. Args: index (int): Index of the sample data to get. Returns: dic
menet/datasets/lyft_dataset.py:66
↓ 3 callersFunctionnuscenes_data_prep
Prepare data related to nuScenes dataset. Related data consists of '.pkl' files recording basic infos, 2D annotations and groundtruth databas
tools/create_data.py:8
↓ 2 callersMethod__init__
( self, load_decoder_from, load_head_from, if_freeze_decoder=True, if_
menet/utils/custom_init.py:112
↓ 2 callersMethod__init__
( self, use_h, use_w, max_epoch, rotate=1, offset=False,
menet/datasets/pipelines/transforms_3d.py:523
↓ 2 callersMethod__init__
(self, sampled_list, name=None, epoch=None,
menet/datasets/pipelines/dpsampler.py:21
↓ 2 callersMethod__init__
( self, in_channels, reduction_ratio=16, pool_types=['avg',
menet/models/attentions/cbam.py:84
↓ 2 callersMethod__init__
( self, backbone, neck = None, init_cfg=None )
menet/models/decoder/second_decoder.py:8
↓ 2 callersMethod_evaluate_single
Evaluation for a single model in nuScenes protocol. Args: result_path (str): Path of the result file. logger (logging
menet/datasets/nuscenes_dataset.py:410
↓ 2 callersMethod_format_bbox
Convert the results to the standard format. Args: results (list[dict]): Testing results of the dataset. jsonfile_pref
menet/datasets/nuscenes_dataset.py:341
↓ 2 callersMethod_freeze_stages
(self)
menet/models/backbone/swin.py:857
↓ 2 callersMethod_parse_losses
Parse the raw outputs (losses) of the network. Args: losses (dict): Raw output of the network, which usually contain
menet/models/fusion_models/base.py:19
↓ 2 callersFunctioncreate_groundtruth_database
Given the raw data, generate the ground truth database. Args: dataset_class_name (str): Name of the input dataset. data_path (str
tools/data_converter/create_gt_database.py:111
↓ 2 callersMethodformat_results
Format the results to json (standard format for COCO evaluation). Args: results (list[dict]): Testing results of the dataset.
menet/datasets/nuscenes_dataset.py:468
↓ 2 callersFunctionget_2d_boxes
Get the 2D annotation records for a given `sample_data_token`. Args: sample_data_token (str): Sample data token belonging to a camera \
tools/data_converter/nuscenes_converter.py:422
↓ 2 callersFunctionget_available_scenes
Get available scenes from the input nuscenes class. Given the raw data, get the information of available scenes for further info generation.
tools/data_converter/nuscenes_converter.py:122
↓ 2 callersMethodhead_loss
The loss() interfaces of the heads in the mmdetection3d are not unified. This is an adaptor.
menet/models/fusion_models/bevfusion.py:86
↓ 2 callersMethodinit_weights
(self)
menet/models/backbone/swin.py:297
↓ 2 callersFunctionlyft_data_prep
Prepare data related to Lyft dataset. Related data consists of '.pkl' files recording basic infos. Although the ground truth database and 2D
tools/create_data.py:54
↓ 2 callersMethodrandom_flip_map_mask
(self, input_dict, direction='horizontal')
menet/datasets/pipelines/transforms_3d.py:498
↓ 2 callersMethodsample
Sample specific number of ground truths. Args: num (int): Sampled number. Returns: list[dict]: Sampled groun
menet/datasets/pipelines/dpsampler.py:64
↓ 2 callersMethodwindow_partition
Args: x: (B, H, W, C) window_size (int): window size Returns: windows: (num_windows*B, window_siz
menet/models/backbone/swin.py:496
↓ 1 callersMethod__call__
(self, module: nn.Module)
menet/utils/custom_init.py:126
↓ 1 callersMethod__call__
(self, results)
menet/datasets/pipelines/transforms_3d.py:555
↓ 1 callersMethod__init__
(self, in_channels, out_channels, mid_channels,
menet/models/utils/edge_residual.py:35
↓ 1 callersMethod__init__
(self, in_channels, out_channels, scale_factor=4, input_feature_index = (0,2),
menet/models/neck/lss.py:14
↓ 1 callersFunction_fill_trainval_infos
Generate the train/val infos from the raw data. Args: nusc (:obj:`NuScenes`): Dataset class in the nuScenes dataset. train_scenes
tools/data_converter/nuscenes_converter.py:163
↓ 1 callersFunction_fill_trainval_infos
Generate the train/val infos from the raw data. Args: lyft (:obj:`LyftDataset`): Dataset class in the Lyft dataset. train_scenes
tools/data_converter/lyft_converter.py:94
↓ 1 callersMethod_getLyftMapMask
Get map mask in nuscenes dataset Args: loc (list[float]): global location of the lidar. (x,y) angle (float):
menet/datasets/pipelines/loading.py:232
↓ 1 callersMethod_getMapPose
(self, data)
menet/datasets/pipelines/loading.py:175
↓ 1 callersMethod_getNusMapMask
Get map mask in nuscenes dataset Args: loc (list[float]): global location of the lidar. (x,y) angle (float):
menet/datasets/pipelines/loading.py:206
↓ 1 callersFunction_parse_coco_ann_info
(ann_info)
tools/data_converter/create_gt_database.py:32
↓ 1 callersFunction_poly2mask
(mask_ann, img_h, img_w)
tools/data_converter/create_gt_database.py:16
↓ 1 callersMethod_random_scale
Private function to randomly set the scale factor. Args: input_dict (dict): Result dict from loading pipeline. Returns:
menet/datasets/pipelines/transforms_3d.py:246
↓ 1 callersMethod_reset
Reset the index of batchsampler to zero.
menet/datasets/pipelines/dpsampler.py:56
↓ 1 callersMethod_rot_bbox_points
Private function to rotate bounding boxes and points. Args: input_dict (dict): Result dict from loading pipeline. Return
menet/datasets/pipelines/transforms_3d.py:260
↓ 1 callersMethod_rot_scale_trans_map_mask
(self, input_dict)
menet/datasets/pipelines/transforms_3d.py:345
↓ 1 callersMethod_rot_scale_trans_vision_bev
(self, input_dict)
menet/datasets/pipelines/transforms_3d.py:387
↓ 1 callersMethod_sample
Sample specific number of ground truths and return indices. Args: num (int): Sampled number. Returns: list[i
menet/datasets/pipelines/dpsampler.py:39
↓ 1 callersMethod_scale_bbox_points
Private function to scale bounding boxes and points. Args: input_dict (dict): Result dict from loading pipeline. Returns
menet/datasets/pipelines/transforms_3d.py:302
↓ 1 callersMethod_trans_bbox_points
Private function to translate bounding boxes and points. Args: input_dict (dict): Result dict from loading pipeline. Ret
menet/datasets/pipelines/transforms_3d.py:325
↓ 1 callersFunctionbev_pool
(feats, coords, B, D, H, W)
menet/ops/bev_pool/bev_pool.py:83
↓ 1 callersFunctionbuild_attention
(cfg)
menet/models/builder.py:16
↓ 1 callersFunctionbuild_backbone
(cfg)
menet/models/builder.py:19
↓ 1 callersFunctionbuild_decoder
(cfg)
menet/models/builder.py:55
↓ 1 callersFunctionbuild_encoder
(cfg)
menet/models/builder.py:52
↓ 1 callersFunctionbuild_fuser
(cfg)
menet/models/builder.py:49
↓ 1 callersFunctionbuild_head
(cfg)
menet/models/builder.py:40
↓ 1 callersFunctionbuild_neck
(cfg)
menet/models/builder.py:34
↓ 1 callersFunctioncorrect_unfold_norm_order
(x)
menet/models/backbone/swin.py:34
↓ 1 callersFunctioncorrect_unfold_reduction_order
(x)
menet/models/backbone/swin.py:27
↓ 1 callersFunctioncrop_image_patch
(pos_proposals, gt_masks, pos_assigned_gt_inds, org_img)
tools/data_converter/create_gt_database.py:91
↓ 1 callersMethoddouble_step_seq
(step1, len1, step2, len2)
menet/models/backbone/swin.py:343
↓ 1 callersMethodevaluate
Evaluation in nuScenes protocol. Args: results (list[dict]): Testing results of the dataset. metric (str | list[str],
menet/datasets/nuscenes_dataset.py:548
↓ 1 callersFunctionfix_lyft
(root_folder='./data/lyft', version='v1.01')
tools/data_converter/lyft_data_fixer.py:8
↓ 1 callersMethodforward_test
TODO: surpport test time augmentation.
menet/models/fusion_models/base.py:115
↓ 1 callersMethodforward_train
(self, **kwargs)
menet/models/fusion_models/base.py:130
↓ 1 callersFunctiongenerate_record
Generate one 2D annotation record given various informations on top of the 2D bounding box coordinates. Args: ann_rec (dict): Origina
tools/data_converter/nuscenes_converter.py:582
↓ 1 callersMethodget_ann_info
Get annotation info according to the given index. Args: index (int): Index of the annotation data to get. Returns:
menet/datasets/nuscenes_dataset.py:292
↓ 1 callersMethodget_ann_info
Get annotation info according to the given index. Args: index (int): Index of the annotation data to get. Returns:
menet/datasets/lyft_dataset.py:137
↓ 1 callersMethodget_task_detections
Rotate nms for each task. Args: num_class_with_bg (int): Number of classes for the current task. batch_cls_preds (lis
menet/models/head/centerpoint.py:120
↓ 1 callersMethodimg_transform
( self, img, rotation, translation, resize, resize_dims, crop, flip, rotate )
menet/datasets/pipelines/transforms_3d.py:139
↓ 1 callersFunctioninput_constructor
(input_shape, model)
tools/analysis_tools/get_flops.py:45
↓ 1 callersFunctionl2_loss
L2 loss. Args: pred (torch.Tensor): The prediction. target (torch.Tensor): The learning target of the prediction. Returns:
menet/models/losses/l2loss.py:9
↓ 1 callersFunctionlidar_nusc_box_to_global
Convert the box from ego to global coordinate. Args: info (dict): Info for a specific sample data, including the calibration
menet/datasets/nuscenes_dataset.py:639
↓ 1 callersFunctionlogsumexp_2d
(tensor)
menet/models/attentions/cbam.py:50
↓ 1 callersMethodloss
(self, input_data, gt_labels, metas=None)
menet/models/encoder/base_encoder.py:8
↓ 1 callersFunctionlyft_eval
Evaluation API for Lyft dataset. Args: lyft (:obj:`LyftDataset`): Lyft class in the sdk. data_root (str): Root of data for readin
menet/datasets/lyft_dataset.py:13
↓ 1 callersFunctionmain
()
tools/train.py:103
↓ 1 callersFunctionmain
()
tools/test.py:105
↓ 1 callersFunctionmain
()
tools/analysis_tools/benchmark.py:31
↓ 1 callersFunctionmain
()
tools/analysis_tools/get_flops.py:66
↓ 1 callersFunctionmake_cuda_ext
( name, module, sources, sources_cuda=[], extra_args=[], extra_include_path=[] )
setup.py:8
↓ 1 callersFunctionmake_divisible
Make divisible function. This function rounds the channel number down to the nearest value that can be divisible by the divisor. Args:
menet/models/attentions/se_layer.py:8
↓ 1 callersFunctionoutput_to_nusc_box
Convert the output to the box class in the nuScenes. Args: detection (dict): Detection results. - boxes_3d (:obj:`BaseInstan
menet/datasets/nuscenes_dataset.py:593
↓ 1 callersFunctionparse_args
()
tools/train.py:34
↓ 1 callersFunctionparse_args
()
tools/test.py:23
↓ 1 callersFunctionparse_args
()
tools/visualize.py:61
↓ 1 callersFunctionparse_args
()
tools/analysis_tools/benchmark.py:15
↓ 1 callersFunctionparse_args
()
tools/analysis_tools/get_flops.py:16
↓ 1 callersFunctionpost_process_coords
Get the intersection of the convex hull of the reprojected bbox corners and the image canvas, return None if no intersection. Args: c
tools/data_converter/nuscenes_converter.py:548
↓ 1 callersMethodsample_all
Sampling all categories of bboxes. Args: gt_info (dict): - gt_bboxes (np.ndarray): Ground truth bou
menet/datasets/pipelines/dpsampler.py:211
↓ 1 callersMethodsample_augmentation
(self, results)
menet/datasets/pipelines/transforms_3d.py:112
↓ 1 callersMethodsample_class_v2
Sampling specific categories of bounding boxes. Args: name (str): Class of objects to be sampled. num (int): Number o
menet/datasets/pipelines/dpsampler.py:309
↓ 1 callersFunctionsave_res
(path, data)
tools/visualize.py:256
↓ 1 callersMethodset_epoch
(self, epoch)
menet/datasets/pipelines/transforms_3d.py:547
↓ 1 callersMethodset_prob
(self, epoch, max_epoch)
menet/datasets/pipelines/transforms_3d.py:552
↓ 1 callersMethodsimple_test
(self, **kwargs)
menet/models/fusion_models/base.py:126
↓ 1 callersFunctionswin_convert
(ckpt)
menet/models/backbone/swin.py:24
↓ 1 callersMethodupdate_transform
(self, input_dict)
menet/datasets/pipelines/transforms_3d.py:512
next →1–100 of 243, ranked by callers