Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/WangYueFt/detr3d
/ types & classes
Types & classes
33 in github.com/WangYueFt/detr3d
⨍
Functions
250
◇
Types & classes
33
↓ 3 callers
Class
ScanNetData
ScanNet data. Generate scannet infos for scannet_converter. Args: root_path (str): Root path of the raw data. split (str): S
tools/data_converter/scannet_data_utils.py:9
↓ 2 callers
Class
SUNRGBDData
SUNRGBD data. Generate scannet infos for sunrgbd_converter. Args: root_path (str): Root path of the raw data. split (str): S
tools/data_converter/sunrgbd_data_utils.py:59
↓ 2 callers
Class
ScanNetSegData
ScanNet dataset used to generate infos for semantic segmentation task. Args: data_root (str): Root path of the raw data. ann_file
tools/data_converter/scannet_data_utils.py:198
↓ 2 callers
Class
_OSA_module
projects/mmdet3d_plugin/models/backbones/vovnet.py:180
↓ 1 callers
Class
GridMask
projects/mmdet3d_plugin/models/utils/grid_mask.py:69
↓ 1 callers
Class
Hsigmoid
projects/mmdet3d_plugin/models/backbones/vovnet.py:156
↓ 1 callers
Class
S3DISData
S3DIS data. Generate s3dis infos for s3dis_converter. Args: root_path (str): Root path of the raw data. split (str): Set spl
tools/data_converter/s3dis_data_utils.py:9
↓ 1 callers
Class
S3DISSegData
S3DIS dataset used to generate infos for semantic segmentation task. Args: data_root (str): Root path of the raw data. ann_file (
tools/data_converter/s3dis_data_utils.py:151
↓ 1 callers
Class
SUNRGBDInstance
tools/data_converter/sunrgbd_data_utils.py:33
↓ 1 callers
Class
_OSA_stage
projects/mmdet3d_plugin/models/backbones/vovnet.py:233
↓ 1 callers
Class
eSEModule
projects/mmdet3d_plugin/models/backbones/vovnet.py:165
Class
BBox3DL1Cost
BBox3DL1Cost. Args: weight (int | float, optional): loss_weight
projects/mmdet3d_plugin/core/bbox/match_costs/match_cost.py:6
Class
CropMultiViewImage
Crop the image Args: size (tuple, optional): Fixed padding size.
projects/mmdet3d_plugin/datasets/pipelines/transform_3d.py:196
Class
CustomNuScenesDataset
r"""NuScenes Dataset. This datset only add camera intrinsics and extrinsics to the results.
projects/mmdet3d_plugin/datasets/nuscenes_dataset.py:7
Class
DGCNN3DHead
Head of DeformDETR3D. Args: with_box_refine (bool): Whether to refine the reference points in the decoder. Defaults to False.
projects/mmdet3d_plugin/models/dense_heads/dgcnn3d_head.py:18
Class
DGCNNAttn
A warpper for DGCNN-type self-attention. Args: embed_dims (int): The embedding dimension. num_heads (int): Parallel attention head
projects/mmdet3d_plugin/models/utils/dgcnn_attn.py:10
Class
Deformable3DDetrTransformerDecoder
Copy the decoder in DETR transformer. Args: return_intermediate (bool): Whether to return intermediate outputs. coder_norm_cfg (di
projects/mmdet3d_plugin/models/utils/detr.py:25
Class
Detr3D
Detr3D.
projects/mmdet3d_plugin/models/detectors/detr3d.py:11
Class
Detr3DCrossAtten
An attention module used in Detr3d. Args: embed_dims (int): The embedding dimension of Attention. Default: 256. num_h
projects/mmdet3d_plugin/models/utils/detr3d_transformer.py:218
Class
Detr3DHead
Head of Detr3D. Args: with_box_refine (bool): Whether to refine the reference points in the decoder. Defaults to False.
projects/mmdet3d_plugin/models/dense_heads/detr3d_head.py:18
Class
Detr3DTransformer
Implements the Detr3D transformer. Args: as_two_stage (bool): Generate query from encoder features. Default: False. nu
projects/mmdet3d_plugin/models/utils/detr3d_transformer.py:36
Class
Detr3DTransformerDecoder
Implements the decoder in DETR3D transformer. Args: return_intermediate (bool): Whether to return intermediate outputs. coder_norm
projects/mmdet3d_plugin/models/utils/detr3d_transformer.py:143
Class
Grid
projects/mmdet3d_plugin/models/utils/grid_mask.py:6
Class
HorizontalRandomFlipMultiViewImage
projects/mmdet3d_plugin/datasets/pipelines/transform_3d.py:262
Class
HungarianAssigner3D
Computes one-to-one matching between predictions and ground truth. This class computes an assignment between the targets and the predictions b
projects/mmdet3d_plugin/core/bbox/assigners/hungarian_assigner_3d.py:17
Class
NMSFreeCoder
Bbox coder for NMS-free detector. Args: pc_range (list[float]): Range of point cloud. post_center_range (list[float]): Limit of th
projects/mmdet3d_plugin/core/bbox/coders/nms_free_coder.py:9
Class
NormalizeMultiviewImage
Normalize the image. Added key is "img_norm_cfg". Args: mean (sequence): Mean values of 3 channels. std (sequence): Std values
projects/mmdet3d_plugin/datasets/pipelines/transform_3d.py:60
Class
ObjDGCNN
Base class of Multi-modality VoxelNet.
projects/mmdet3d_plugin/models/detectors/obj_dgcnn.py:9
Class
PadMultiViewImage
Pad the multi-view image. There are two padding modes: (1) pad to a fixed size and (2) pad to the minimum size that is divisible by some numbe
projects/mmdet3d_plugin/datasets/pipelines/transform_3d.py:8
Class
PhotoMetricDistortionMultiViewImage
Apply photometric distortion to image sequentially, every transformation is applied with a probability of 0.5. The position of random contrast is
projects/mmdet3d_plugin/datasets/pipelines/transform_3d.py:96
Class
RandomScaleImageMultiViewImage
Random scale the image Args: scales
projects/mmdet3d_plugin/datasets/pipelines/transform_3d.py:224
Class
VoVNet
projects/mmdet3d_plugin/models/backbones/vovnet.py:269
Class
Waymo2KITTI
Waymo to KITTI converter. This class serves as the converter to change the waymo raw data to KITTI format. Args: load_dir (str):
tools/data_converter/waymo_converter.py:23