MCPcopy Create free account

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

Functions243 in github.com/WHU-USI3DV/MENet

↓ 1 callersFunctionvisualize_lidar
( fpath: str, lidar: Optional[np.ndarray] = None, bboxes: Optional[LiDARInstance3DBoxe
tools/visualize.py:75
↓ 1 callersFunctionvisualize_lidar_and_map
( fpath: str, lidar: Optional[np.ndarray], bboxes: Optional[LiDARInstance3DBoxes],
tools/visualize.py:163
↓ 1 callersFunctionvisualize_map
( fpath: str, masks: np.ndarray, classes: List[str], background: Tuple[int, in
tools/visualize.py:144
↓ 1 callersMethodvoxelize
Apply hard voxelization to points.
menet/models/encoder/voxel_encoder.py:41
↓ 1 callersMethodvoxelize_dv
Apply dynamic voxelization to points. Args: points (list[torch.Tensor]): Points of each sample. Returns: tup
menet/models/encoder/voxel_encoder.py:60
↓ 1 callersMethodwindow_reverse
Args: windows: (num_windows*B, window_size, window_size, C) window_size (int): Window size H (int): Heigh
menet/models/backbone/swin.py:479
FunctionPYBIND11_MODULE
menet/ops/bev_pool/src/bev_pool.cpp:89
Method__call__
(self, module: nn.Module)
menet/utils/custom_init.py:52
Method__call__
(self, module: nn.Module)
menet/utils/custom_init.py:103
Method__call__
Call function to sample ground truth objects to the data. Args: data (dict): Result dict from loading pipeline. Returns:
menet/datasets/pipelines/transforms_3d.py:28
Method__call__
(self, data: Dict[str, Any])
menet/datasets/pipelines/transforms_3d.py:171
Method__call__
(self, data: Dict[str, Any])
menet/datasets/pipelines/transforms_3d.py:210
Method__call__
Private function to rotate, scale and translate bounding boxes and points. Args: input_dict (dict): Result dict from load
menet/datasets/pipelines/transforms_3d.py:398
Method__call__
Call function to flip points, values in the ``bbox3d_fields`` and also flip 2D image and its annotations. Args: input_dic
menet/datasets/pipelines/transforms_3d.py:452
Method__call__
Call function to load multi-view image from files. Args: results (dict): Result dict containing multi-view image filenames.
menet/datasets/pipelines/loading.py:28
Method__call__
(self, data: Dict[str, Any])
menet/datasets/pipelines/loading.py:154
Method__call__
Call function to transform and format common fields in results. Args: results (dict): Result dict contains the data to convert.
menet/datasets/pipelines/formating.py:12
Method__call__
Call function to collect keys in results. The keys in ``meta_keys`` will be converted to :obj:`mmcv.DataContainer`. Args:
menet/datasets/pipelines/formating.py:166
Method__init__
( self, load_multiview_encoder_from=None, load_lidar_encoder_from=None,
menet/utils/custom_init.py:8
Method__init__
( self, if_freeze=False, **kwargs )
menet/utils/custom_init.py:95
Method__init__
(self, modules: Optional[dict] = None, init_cfg: Optional[dict] = None)
menet/runner/base_module.py:14
Method__init__
( self, ann_file, pipeline=None, data_root=None, classes=None,
menet/datasets/nuscenes_dataset.py:125
Method__init__
(self, db_sampler, stop_epoch=None, sample_2d=False)
menet/datasets/pipelines/transforms_3d.py:20
Method__init__
( self, is_train, final_dim=[256, 704], resize_lim=(-0.06, 0.11), bo
menet/datasets/pipelines/transforms_3d.py:94
Method__init__
(self, mean, std)
menet/datasets/pipelines/transforms_3d.py:201
Method__init__
(self, rot_range=[-0.78539816, 0.78539816], scale_ratio_range=[0.95, 1.05],
menet/datasets/pipelines/transforms_3d.py:217
Method__init__
(self, sync_2d=True, flip_ratio_bev_horizontal=0.0, flip_ra
menet/datasets/pipelines/transforms_3d.py:434
Method__init__
(self, to_float32=False, color_type="unchanged")
menet/datasets/pipelines/loading.py:24
Method__init__
( self, data_root: str, xbound: Tuple[float, float, float], ybound: Tuple[floa
menet/datasets/pipelines/loading.py:90
Method__init__
(self, info_path, data_root, rate, prepare
menet/datasets/pipelines/dpsampler.py:93
Method__init__
( self, info_path, data_root, rate, prepare, sample_groups,
menet/datasets/pipelines/dpsampler.py:372
Method__init__
( self, input_data_keys=["img", "map_mask"], gt_keys=[], meta_keys=['filename'
menet/datasets/pipelines/formating.py:151
Method__init__
(self, in_channels=3, embed_dims=768, conv_type=None,
menet/models/backbone/swin.py:99
Method__init__
(self, in_channels, out_channels, stride=2,
menet/models/backbone/swin.py:189
Method__init__
(self, embed_dims, num_heads, window_size,
menet/models/backbone/swin.py:373
Method__init__
(self, embed_dims, num_heads, feedforward_channels,
menet/models/backbone/swin.py:535
Method__init__
(self, embed_dims, num_heads, feedforward_channels,
menet/models/backbone/swin.py:619
Method__init__
(self, pretrain_img_size=224, in_channels=3, embed_dims=96,
menet/models/backbone/swin.py:730
Method__init__
(self, numC_input, numC_middle=None, num_layer=[2,2,2],
menet/models/backbone/resnet.py:14
Method__init__
( self, conv_group_cfg = dict( type="EdgeResidualGroup",
menet/models/utils/spatial_transformer.py:16
Method__init__
(self, in_channels, out_channels, kernel_size=1,
menet/models/utils/downsample.py:23
Method__init__
( self, input_channels, out_channels=[32, 64], mid_channels=[
menet/models/utils/edge_residual.py:117
Method__init__
( self, voxel_layer, voxel_encoder, middle_encoder, stream_name = "lid
menet/models/encoder/voxel_encoder.py:11
Method__init__
(self, stream_name="base", init_cfg=None)
menet/models/encoder/base_encoder.py:4
Method__init__
(self, stream_name="map", init_cfg=None)
menet/models/encoder/map_encoder.py:18
Method__init__
(self, output_size, mode = "bilinear", stream_name="map", init_cfg=None)
menet/models/encoder/map_encoder.py:28
Method__init__
( self, conv_group_cfg = dict( type="EdgeResidualGroup",
menet/models/encoder/map_encoder.py:65
Method__init__
(self, init_cfg=None)
menet/models/fusion_models/base.py:15
Method__init__
( self, encoders: List[Dict], fuser: Dict[str, Any], decoder: Dict[str, Any],
menet/models/fusion_models/bevfusion.py:25
Method__init__
( self, gate_channel, reduction_ratio=16, n
menet/models/attentions/bam.py:14
Method__init__
( self, gate_channel, reduction_ratio=16, d
menet/models/attentions/bam.py:40
Method__init__
( self, gate_channel, reduction_ratio=16, di
menet/models/attentions/bam.py:65
Method__init__
(self, in_channels, sync_bn=False)
menet/models/attentions/bam.py:120
Method__init__
(self, in_channels, squeeze_channels=None, ratio=16,
menet/models/attentions/se_layer.py:57
Method__init__
(self, gate_channels, reduction_ratio=16, pool_types=['avg', 'max'])
menet/models/attentions/cbam.py:14
Method__init__
( self, conv_cfg=dict(type='Conv2d'), norm_cfg=dict(type='BN', eps=1e-5, m
menet/models/attentions/cbam.py:61
Method__init__
( self, backbone, neck = None, init_cfg=None )
menet/models/decoder/second_decoder.py:38
Method__init__
( self, backbone, neck = None, init_cfg=None )
menet/models/decoder/second_decoder.py:48
Method__init__
(self, in_channels, out_channels, num_outs,
menet/models/neck/fpn.py:61
Method__init__
( self, in_channels, out_channels, start_level=0, end_level=-1,
menet/models/neck/lss.py:74
Method__init__
(self, reduction='mean', loss_weight=1.0)
menet/models/losses/l2loss.py:33
Method__init__
( self, in_channels, out_channels=256, attention_only=False, #
menet/models/fuser/attention_fuser.py:21
Method__init__
(self, init_cfg=None)
menet/models/fuser/concat_fuser.py:7
Method__repr__
str: Return a string that describes the module.
menet/datasets/pipelines/loading.py:70
Method__repr__
str: Return a string that describes the module.
menet/datasets/pipelines/formating.py:201
Method_evaluate_single
Evaluation for a single model in Lyft protocol. Args: result_path (str): Path of the result file. logger (logging.Log
menet/datasets/lyft_dataset.py:200
Method_inner_forward
(x)
menet/models/utils/edge_residual.py:91
Methodbackward
(ctx, gradx, gradgeom)
menet/ops/bev_pool/bev_pool.py:27
Methodbackward
(ctx, out_grad)
menet/ops/bev_pool/bev_pool.py:64
Functionbev_pool_backward
Function: pillar pooling (backward, cuda) Args: out_grad : input features, FloatTensor[b, d, h, w, c] geom_feats : input coord
menet/ops/bev_pool/src/bev_pool.cpp:60
Functionbev_pool_forward
Function: pillar pooling (forward, cuda) Args: x : input features, FloatTensor[n, c] geom_feats : input coordinates, In
menet/ops/bev_pool/src/bev_pool.cpp:22
Functionbuild_kd_loss
(cfg)
menet/models/builder.py:31
Functionbuild_loss
(cfg)
menet/models/builder.py:46
Functionbuild_losses
(cfg)
menet/models/builder.py:25
Functioncreate_lyft_infos
Create info file of lyft dataset. Given the raw data, generate its related info file in pkl format. Args: root_path (str): Path of t
tools/data_converter/lyft_converter.py:19
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
tools/data_converter/nuscenes_converter.py:42
Methodcrop_image
(image: np.array, pixel_coords: list, size: int)
menet/datasets/pipelines/loading.py:239
Functioncrop_image_patch_v2
(pos_proposals, pos_assigned_gt_inds, gt_masks)
tools/data_converter/create_gt_database.py:68
Methodevaluate_map
(self, results)
menet/datasets/nuscenes_dataset.py:513
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
tools/data_converter/nuscenes_converter.py:356
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
tools/data_converter/lyft_converter.py:217
Methodfilter_by_difficulty
Filter ground truths by difficulties. Args: db_infos (dict): Info of groundtruth database. removed_difficulty (list):
menet/datasets/pipelines/dpsampler.py:171
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
menet/datasets/pipelines/dpsampler.py:190
Functionfix_bn
(module)
menet/utils/weights_utils.py:6
Methodforward
(ctx, x, geom_feats, ranks)
menet/ops/bev_pool/bev_pool.py:10
Methodforward
(ctx, x, geom_feats, ranks, B, D, H, W)
menet/ops/bev_pool/bev_pool.py:39
Methodforward
(self, x)
menet/models/backbone/swin.py:149
Methodforward
x: x.shape -> [B, H*W, C] hw_shape: (H, W)
menet/models/backbone/swin.py:213
Methodforward
Args: x (tensor): input features with shape of (num_windows*B, N, C) mask (tensor | None, Optional): mask with shape
menet/models/backbone/swin.py:300
Methodforward
(self, query, hw_shape)
menet/models/backbone/swin.py:402
Methodforward
(self, x)
menet/models/backbone/swin.py:579
Methodforward
(self, x, hw_shape)
menet/models/backbone/swin.py:664
Methodforward
(self, x)
menet/models/backbone/swin.py:943
Methodforward
(self, x)
menet/models/backbone/resnet.py:62
Methodforward
(self, x)
menet/models/utils/spatial_transformer.py:50
Methodforward
(self, x)
menet/models/utils/downsample.py:42
Methodforward
(self, x)
menet/models/utils/edge_residual.py:89
Methodforward
(self, x)
menet/models/utils/edge_residual.py:164
Methodforward
Forward pass. Args: feats (list[torch.Tensor]): Multi-level features, e.g., features produced by FPN. Retu
menet/models/head/centerpoint.py:11
Methodforward
(self, data, metas=None)
menet/models/encoder/voxel_encoder.py:24
← previousnext →101–200 of 243, ranked by callers