Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/OpenDriveLab/OccNet
/ types & classes
Types & classes
44 in github.com/OpenDriveLab/OccNet
⨍
Functions
203
◇
Types & classes
44
↳
Endpoints
1
↓ 4 callers
Class
EgoPoseDataset
projects/mmdet3d_plugin/datasets/ego_pose_extractor.py:14
↓ 2 callers
Class
_OSA_module
projects/mmdet3d_plugin/models/backbones/vovnet.py:180
↓ 1 callers
Class
EgoPoseDataset
tools/ray_iou/ego_pose_extractor.py:14
↓ 1 callers
Class
GridMask
projects/mmdet3d_plugin/models/utils/grid_mask.py:70
↓ 1 callers
Class
Hsigmoid
projects/mmdet3d_plugin/models/backbones/vovnet.py:156
↓ 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
AdamW2
r"""Implements AdamW algorithm. Solve the bug of torch 1.8 The original Adam algorithm was proposed in `Adam: A Method for Stochastic Optimizatio
projects/mmdet3d_plugin/models/opt/adamw.py:11
Class
BBox3DL1Cost
BBox3DL1Cost. Args: weight (int | float, optional): loss_weight
projects/mmdet3d_plugin/core/bbox/match_costs/match_cost.py:6
Class
BEVFormerEncoder
Attention with both self and cross Implements the decoder in DETR transformer. Args: return_intermediate (bool): Whether to r
projects/mmdet3d_plugin/bevformer/modules/encoder.py:29
Class
BEVFormerLayer
Implements decoder layer in DETR transformer. Args: attn_cfgs (list[`mmcv.ConfigDict`] | list[dict] | dict )): Configs for
projects/mmdet3d_plugin/bevformer/modules/encoder.py:243
Class
BEVFormerOcc
BEVFormer. Args: video_test_mode (bool): Decide whether to use temporal information during inference.
projects/mmdet3d_plugin/bevformer/detectors/bevformer_occ.py:21
Class
BEVFormerOccHead
Head of Detr3D. Args: with_box_refine (bool): Whether to refine the reference points in the decoder. Defaults to False.
projects/mmdet3d_plugin/bevformer/dense_heads/bevformer_occ_head.py:33
Class
CustomCollect3D
Collect data from the loader relevant to the specific task. This is usually the last stage of the data loader pipeline. Typically keys is set
projects/mmdet3d_plugin/datasets/pipelines/transform_3d.py:204
Class
CustomDefaultFormatBundle3D
Default formatting bundle. It simplifies the pipeline of formatting common fields for voxels, including "proposals", "gt_bboxes", "gt_labels
projects/mmdet3d_plugin/datasets/pipelines/formating.py:13
Class
CustomDistEvalHook
projects/mmdet3d_plugin/core/evaluation/eval_hooks.py:29
Class
CustomMSDeformableAttention
An attention module used in Deformable-Detr. `Deformable DETR: Deformable Transformers for End-to-End Object Detection. <https://arxiv.org
projects/mmdet3d_plugin/bevformer/modules/decoder.py:133
Class
DetectionTransformerDecoder
Implements the decoder in DETR3D transformer. Args: return_intermediate (bool): Whether to return intermediate outputs. coder_n
projects/mmdet3d_plugin/bevformer/modules/decoder.py:53
Class
DistributedGroupSampler
Sampler that restricts data loading to a subset of the dataset. It is especially useful in conjunction with :class:`torch.nn.parallel.Distribu
projects/mmdet3d_plugin/datasets/samplers/group_sampler.py:15
Class
DistributedSampler
projects/mmdet3d_plugin/datasets/samplers/distributed_sampler.py:9
Class
EpochBasedRunner_video
# basic logic input_sequence = [a, b, c] # given a sequence of samples prev_bev = None for each in input_sequcene[:-
projects/mmdet3d_plugin/bevformer/runner/epoch_based_runner.py:19
Class
GradChecker
projects/mmdet3d_plugin/models/hooks/hooks.py:6
Class
Grid
projects/mmdet3d_plugin/models/utils/grid_mask.py:7
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
LearnedPositionalEncoding3D
Position embedding with learnable embedding weights. Args: num_feats (int): The feature dimension for each position along
projects/mmdet3d_plugin/models/utils/positional_encoding.py:10
Class
LoadOccGTFromFile
Load multi channel images from a list of separate channel files. Expects results['img_filename'] to be a list of filenames. note that we r
projects/mmdet3d_plugin/datasets/pipelines/loading.py:7
Class
MSDeformableAttention3D
An attention module used in BEVFormer based on Deformable-Detr. `Deformable DETR: Deformable Transformers for End-to-End Object Detection. <
projects/mmdet3d_plugin/bevformer/modules/spatial_cross_attention.py:179
Class
MultiScaleDeformableAttnFunction_fp16
projects/mmdet3d_plugin/bevformer/modules/multi_scale_deformable_attn_function.py:15
Class
MultiScaleDeformableAttnFunction_fp32
projects/mmdet3d_plugin/bevformer/modules/multi_scale_deformable_attn_function.py:90
Class
MyCustomBaseTransformerLayer
Base `TransformerLayer` for vision transformer. It can be built from `mmcv.ConfigDict` and support more flexible customization, for example,
projects/mmdet3d_plugin/bevformer/modules/custom_base_transformer_layer.py:38
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:10
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:66
Class
NuSceneOcc
r"""NuScenes Dataset. This datset only add camera intrinsics and extrinsics to the results.
projects/mmdet3d_plugin/datasets/nuscenes_occ.py:22
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:12
Class
PerceptionTransformer
Implements the Detr3D transformer. Args: as_two_stage (bool): Generate query from encoder features. Default: False.
projects/mmdet3d_plugin/bevformer/modules/transformer.py:27
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:103
Class
RandomScaleImageMultiViewImage
Random scale the image Args: scales
projects/mmdet3d_plugin/datasets/pipelines/transform_3d.py:293
Class
RelPositionEmbedding
projects/mmdet3d_plugin/models/utils/position_embedding.py:5
Class
SpatialCrossAttention
An attention module used in BEVFormer. Args: embed_dims (int): The embedding dimension of Attention. Default: 256.
projects/mmdet3d_plugin/bevformer/modules/spatial_cross_attention.py:32
Class
TemporalSelfAttention
An attention module used in BEVFormer based on Deformable-Detr. `Deformable DETR: Deformable Transformers for End-to-End Object Detection.
projects/mmdet3d_plugin/bevformer/modules/temporal_self_attention.py:26
Class
TransferWeight
projects/mmdet3d_plugin/bevformer/hooks/custom_hooks.py:6
Class
TransformerOcc
Implements the Detr3D transformer. Args: as_two_stage (bool): Generate query from encoder features. Default: False.
projects/mmdet3d_plugin/bevformer/modules/transformer_occ.py:27
Class
VoVNet
projects/mmdet3d_plugin/models/backbones/vovnet.py:269
Class
nuScenesDataset
projects/mmdet3d_plugin/datasets/nuscenes_ego_pose_loader.py:13