MCPcopy Create free account

hub / github.com/cxmomo/RaCFormer / types & classes

Types & classes81 in github.com/cxmomo/RaCFormer

↓ 4 callersClass_ASPPModule
models/necks/view_transformer_racformer.py:329
↓ 2 callersClassBEVSampling
Adaptive Spatio-temporal Sampling
models/racformer_transformer.py:425
↓ 2 callersClassMlp
models/necks/view_transformer_racformer.py:439
↓ 2 callersClassSELayer
models/necks/view_transformer_racformer.py:465
↓ 2 callersClass_OSA_module
models/backbones/vovnet.py:181
↓ 1 callersClassASPP
models/necks/view_transformer_racformer.py:363
↓ 1 callersClassAdaptiveMixing
Adaptive Mixing
models/racformer_transformer.py:545
↓ 1 callersClassBEVSelfAttention
An attention module used in BEVFormer based on Deformable-Detr. `Deformable DETR: Deformable Transformers for End-to-End Object Detection.
models/bev_self_attention.py:22
↓ 1 callersClassConvGRU
models/racformer_transformer.py:661
↓ 1 callersClassConvGRUCell
models/racformer_transformer.py:691
↓ 1 callersClassDepthNet
models/necks/view_transformer_racformer.py:481
↓ 1 callersClassDumpConfig
models/utils.py:308
↓ 1 callersClassFlashAttention
Implement the scaled dot product attention with softmax. Arguments --------- softmax_scale: The temperature to use for the softmax att
models/model_utils/attention.py:34
↓ 1 callersClassFocalLoss
r"""Criterion that computes Focal loss. According to :cite:`lin2018focal`, the Focal loss is computed as follows: .. math:: \text{FL}(
models/necks/focalloss.py:138
↓ 1 callersClassGpuPhotoMetricDistortion
Apply photometric distortion to image sequentially, every transformation is applied with a probability of 0.5. The position of random contrast is
models/utils.py:219
↓ 1 callersClassGridMask
models/utils.py:8
↓ 1 callersClassHolder
models/checkpoint.py:371
↓ 1 callersClassHsigmoid
models/backbones/vovnet.py:157
↓ 1 callersClassModelEMA
Model Exponential Moving Average from https://github.com/rwightman/ pytorch-image-models Keep a moving average of everything in the model stat
models/hook/ema.py:17
↓ 1 callersClassRaCFormerSampling
Adaptive Spatio-temporal Sampling
models/racformer_transformer.py:334
↓ 1 callersClassRaCFormerTransformerDecoder
models/racformer_transformer.py:59
↓ 1 callersClassRaCFormerTransformerDecoderLayer
models/racformer_transformer.py:141
↓ 1 callersClassRadarBEVTemporalEncoder
Adaptive Spatio-temporal Sampling
models/racformer_transformer.py:614
↓ 1 callersClassScaleAdaptiveSelfAttention
Scale-adaptive Self Attention
models/racformer_transformer.py:278
↓ 1 callersClassVersion
models/utils.py:320
↓ 1 callersClass_OSA_stage
models/backbones/vovnet.py:241
↓ 1 callersClasseSEModule
models/backbones/vovnet.py:166
ClassBBox3DL1Cost
BBox3DL1Cost. Args: weight (int | float, optional): loss_weight
models/bbox/match_costs/match_cost.py:6
ClassBBoxBEVL1Cost
models/bbox/match_costs/match_cost.py:67
ClassBinaryFocalLossWithLogits
r"""Criterion that computes Focal loss. According to :cite:`lin2018focal`, the Focal loss is computed as follows: .. math:: \text{FL}(
models/necks/focalloss.py:248
ClassCheck_CheckpointHook
Save checkpoints periodically. Args: interval (int): The saving period. If ``by_epoch=True``, interval indicates epochs, othe
models/hook/check_checkpoint.py:15
ClassCheckpointFunction
models/checkpoint.py:73
ClassCustomFPN
r"""Feature Pyramid Network. LSSFPN: 1/16 + 1/32 -> lateral -> 1/32 upsample to 1/16 + 1/16 -> conv This is an implementation of paper `F
models/necks/fpn.py:11
ClassCustomNuScenesDataset
loaders/nuscenes_dataset.py:21
ClassCustomNuScenesDataset_radar
loaders/nuscenes_dataset.py:106
ClassCustomResNet
Args: num_layer (List[int]): ResBlock for each layer num_channels (List[int]): Output channel number for each stage
models/backbones/resnet.py:11
ClassFPN_LSS
models/necks/lss_fpn.py:11
ClassFlashMHA
models/model_utils/attention.py:99
ClassGlobalRotScaleTransImage
loaders/pipelines/transforms.py:346
ClassHisInfoControlHook
models/hook/sequentialsontrol.py:31
ClassHungarianAssigner3D
models/bbox/assigners/hungarian_assigner_3d.py:16
ClassISSaveHook
EMAHook used in BEVDepth. Modified from https://github.com/Megvii-Base Detection/BEVDepth/blob/main/callbacks/ema.py.
models/hook/is_save.py:19
ClassIoU3DCost
models/bbox/match_costs/match_cost.py:83
ClassLSSViewTransformerBEVDepth_racformer
models/necks/view_transformer_racformer.py:572
ClassLSSViewTransformer_racformer
r"""Lift-Splat-Shoot view transformer with BEVPoolv2 implementation. Please refer to the `paper <https://arxiv.org/abs/2008.05711>`_ and
models/necks/view_transformer_racformer.py:17
ClassLoadMultiViewImageFromMultiSweeps
loaders/pipelines/loading.py:599
ClassLoadMultiViewImageFromMultiSweepsFuture
loaders/pipelines/loading.py:42
ClassLoadPointsFromFile
Load Points From File. Load points from file. Args: coord_type (str): The type of coordinates of points cloud. Available
loaders/pipelines/loading.py:146
ClassLoadVoDPointsFromFile
Load Points From File. Load points from file. Args: coord_type (str): The type of coordinates of points cloud. Available
loaders/pipelines/loading.py:271
ClassLoadnuradarpoints
Load radar Points From File. Load sunrgbd and scannet points from file. Args: coord_type (str): The type of coordinates of points cl
loaders/pipelines/loading.py:728
ClassLoadradarpointsFromMultiSweeps
loaders/pipelines/loading.py:806
ClassMEGVIIEMAHook
EMAHook used in BEVDepth. Modified from https://github.com/Megvii-Base Detection/BEVDepth/blob/main/callbacks/ema.py.
models/hook/ema.py:63
ClassMSMVSamplingC2345
models/csrc/wrapper.py:78
ClassMSMVSamplingC23456
models/csrc/wrapper.py:122
ClassMSMVSamplingC45
models/csrc/wrapper.py:100
ClassMultiScaleDeformableAttnFunction_fp16
models/multi_scale_deformable_attn_function.py:15
ClassMultiScaleDeformableAttnFunction_fp32
models/multi_scale_deformable_attn_function.py:90
ClassMyTensorboardLoggerHook
utils.py:136
ClassMyTextLoggerHook
utils.py:43
ClassNMSFreeCoder
Bbox coder for NMS-free detector. Args: pc_range (list[float]): Range of point cloud. post_center_range (list[float]): Limit of th
models/bbox/coders/nms_free_coder.py:9
ClassNormalizeMultiviewImage
Normalize the image. Added key is "img_norm_cfg". Args: mean (sequence): Mean values of 3 channels. std (sequence): Std values
loaders/pipelines/transforms.py:70
ClassPadMultiViewImage
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
loaders/pipelines/transforms.py:11
ClassPhotoMetricDistortionMultiViewImage
Apply photometric distortion to image sequentially, every transformation is applied with a probability of 0.5. The position of random contrast is
loaders/pipelines/transforms.py:118
ClassPointToMultiViewDepth
loaders/pipelines/loading.py:465
ClassPolarHungarianAssigner3D
models/bbox/assigners/polar_hungarian_assigner_3d.py:16
ClassQuickCumsumCuda
r"""BEVPoolv2 implementation for Lift-Splat-Shoot view transformation. Please refer to the `paper <https://arxiv.org/abs/2211.17111>`_
models/csrc/bev_pool_v2/bev_pool.py:11
ClassRaCFormatBundle3D
Default formatting bundle. It simplifies the pipeline of formatting common fields for voxels, including "proposals", "gt_bboxes", "gt_labels"
loaders/pipelines/formatng.py:10
ClassRaCFormer
models/racformer.py:19
ClassRaCFormerTransformer
models/racformer_transformer.py:18
ClassRaCFormer_head
models/racformer_head.py:14
ClassRaCGlobalRotScaleTransImage
loaders/pipelines/transforms.py:398
ClassRadarPointCloud_v2
loaders/nuscenes_dataset.py:237
ClassRadarPointToMultiViewDepth
loaders/pipelines/loading.py:514
ClassRandomFlip
Flip the image & bbox & mask. If the input dict contains the key "flip", then the flag will be used, otherwise it will be randomly decided by
loaders/pipelines/transforms.py:503
ClassRandomTransformImage
loaders/pipelines/transforms.py:220
ClassSequentialControlHook
models/hook/sequentialsontrol.py:9
ClassTRTBEVPoolv2
models/csrc/bev_pool_v2/bev_pool.py:95
ClassThetaL1Cost
BBox3DL1Cost. Args: weight (int | float, optional): loss_weight
models/bbox/match_costs/match_cost.py:30
ClassTransformDepthmap
Normalize the image. Added key is "img_norm_cfg". Args: mean (sequence): Mean values of 3 channels. std (sequence): Std valu
loaders/pipelines/transforms.py:468
ClassVoDMonoDataset
Monocular 3D detection on VoD Dataset. Args: data_root (str): Path of dataset root. info_file (str): Path of info file. l
loaders/vod_mono_dataset.py:17
ClassVoVNet
models/backbones/vovnet.py:276