Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/Song-Jingyu/LiRaFusion
/ types & classes
Types & classes
46 in github.com/Song-Jingyu/LiRaFusion
⨍
Functions
247
◇
Types & classes
46
↓ 4 callers
Class
UnaryBlock
plugin/lirafusion/models/backbones/models/blocks.py:469
↓ 3 callers
Class
BatchNormBlock
plugin/lirafusion/models/backbones/models/blocks.py:429
↓ 3 callers
Class
KPConv
plugin/lirafusion/models/backbones/models/blocks.py:143
↓ 1 callers
Class
GlobalAverageBlock
plugin/lirafusion/models/backbones/models/blocks.py:651
↓ 1 callers
Class
GridMask
plugin/lirafusion/models/utils/grid_mask.py:69
↓ 1 callers
Class
MaxPoolBlock
plugin/lirafusion/models/backbones/models/blocks.py:682
↓ 1 callers
Class
NearestUpsampleBlock
plugin/lirafusion/models/backbones/models/blocks.py:664
↓ 1 callers
Class
RadarPoints
Points of instances in LIDAR coordinates. Args: tensor (torch.Tensor | np.ndarray | list): a N x points_dim matrix. points_dim (i
plugin/lirafusion/datasets/radar_points.py:19
↓ 1 callers
Class
ResnetBottleneckBlock
plugin/lirafusion/models/backbones/models/blocks.py:563
↓ 1 callers
Class
SimpleBlock
plugin/lirafusion/models/backbones/models/blocks.py:506
Class
BBox3DL1Cost
BBox3DL1Cost. Args: weight (int | float, optional): loss_weight
plugin/lirafusion/core/bbox/match_costs/match_cost.py:6
Class
CenterPoint_ezfusion_lr
Base class of Multi-modality VoxelNet.
plugin/lirafusion/models/detectors/CenterPoint_ezfusion_lr.py:21
Class
CenterPoint_voxel_fusion
Base class of Multi-modality VoxelNet.
plugin/lirafusion/models/detectors/centerpoint_voxel_fusion.py:17
Class
CenterPoint_voxel_fusion_middle
Base class of Multi-modality VoxelNet.
plugin/lirafusion/models/detectors/centerpoint_voxel_fusion_middle.py:27
Class
CropMultiViewImage
Crop the image Args: size (tuple, optional): Fixed padding size.
plugin/lirafusion/datasets/transform_3d.py:196
Class
DeformableFUTR3DHead
Head of DeformDETR3DCam. Args: with_box_refine (bool): Whether to refine the reference points in the decoder. Defaults to
plugin/lirafusion/models/dense_head/detr_mdfs_head.py:20
Class
FPNV2
r"""Feature Pyramid Network. This is an implementation of paper `Feature Pyramid Networks for Object Detection <https://arxiv.org/abs/1612.031
plugin/lirafusion/models/necks/fpn.py:10
Class
FUTR3D
Base class of Multi-modality VoxelNet.
plugin/lirafusion/models/detectors/futr3d.py:25
Class
FUTR3DCrossAtten
An attention module used in Deformable-Detr. `Deformable DETR: Deformable Transformers for End-to-End Object Detection. <https://arxiv.org/p
plugin/lirafusion/models/utils/attention.py:44
Class
FUTR3DTransformer
Implements the DeformableDETR transformer. Args: num_feature_levels (int): Number of feature maps from FPN: Default: 4.
plugin/lirafusion/models/utils/transformer.py:42
Class
FUTR3DTransformerDecoder
Implements the decoder in DETR transformer. Args: return_intermediate (bool): Whether to return intermediate outputs. coder_norm_c
plugin/lirafusion/models/utils/transformer.py:165
Class
FadeOjectSampleHook
Switch the mode of YOLOX during training. This hook turns off the mosaic and mixup data augmentation and switches to use L1 loss in bbox_hea
plugin/lirafusion/core/fade_hook.py:6
Class
GlobalRotScaleTrans_radar
Apply global rotation, scaling and translation to a 3D scene. - adding support for both lidar and radar points Args: rot_range (list[floa
plugin/lirafusion/datasets/radar_points.py:115
Class
Grid
plugin/lirafusion/models/utils/grid_mask.py:6
Class
HorizontalRandomFlipMultiViewImage
plugin/lirafusion/datasets/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
plugin/lirafusion/core/bbox/assigners/hungarian_assigner_3d.py:17
Class
Lidar_Radar_middle_fusion
Fuse feature maps from lidar and radar
plugin/lirafusion/models/backbones/middle_fusion_layer.py:19
Class
LoadRadarPointsMultiSweeps
Load radar points from multiple sweeps. This is usually used for nuScenes dataset to utilize previous sweeps. Args: sweeps_num (int
plugin/lirafusion/datasets/loading.py:572
Class
LoadRadarPointsMultiSweeps_lirafusion
Load radar points from multiple sweeps. This is usually used for nuScenes dataset to utilize previous sweeps. Args: sweeps_num (int
plugin/lirafusion/datasets/loading.py:735
Class
LoadReducedPointsFromFile
Load Points From File. Load sunrgbd and scannet points from file. Args: coord_type (str): The type of coordinates of points cloud.
plugin/lirafusion/datasets/loading.py:67
Class
LoadReducedPointsFromMultiSweeps
Load points from multiple sweeps. This is usually used for nuScenes dataset to utilize previous sweeps. Args: sweeps_num (int): Num
plugin/lirafusion/datasets/loading.py:190
Class
MultiScaleFlipAug3D_radar
Test-time augmentation with multiple scales and flipping. Args: transforms (list[dict]): Transforms to apply in each augmentation.
plugin/lirafusion/datasets/radar_points.py:431
Class
NMSFreeCoder
Bbox coder for NMS-free detector. Args: pc_range (list[float]): Range of point cloud. post_center_range (list[float]): Limit of
plugin/lirafusion/core/bbox/coders/nms_free_coder.py:9
Class
NMSFreeCoderQuery
Bbox coder for NMS-free detector. Args: pc_range (list[float]): Range of point cloud. post_center_range (list[float]): Limit of
plugin/lirafusion/core/bbox/coders/nms_free_coder.py:117
Class
NormalizeMultiviewImage
Normalize the image. Added key is "img_norm_cfg". Args: mean (sequence): Mean values of 3 channels. std (sequence): Std va
plugin/lirafusion/datasets/transform_3d.py:60
Class
NuScenesDatasetRadar
r"""NuScenes Dataset. This class serves as the API for experiments on the NuScenes Dataset. Please refer to `NuScenes Dataset <https://www.n
plugin/lirafusion/datasets/nuscenes_radar.py:16
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 num
plugin/lirafusion/datasets/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
plugin/lirafusion/datasets/transform_3d.py:96
Class
PointGlobalRotScaleTrans
Apply global rotation, scaling and translation to a 3D scene. Args: rot_range (list[float]): Range of rotation angle. Def
plugin/lirafusion/datasets/transform_3d.py:416
Class
PointRandomFlip3D
Flip the points & bbox. If the input dict contains the key "flip", then the flag will be used, otherwise it will be randomly decided by a
plugin/lirafusion/datasets/transform_3d.py:309
Class
Radar_format
Get radar points to new radar points.
plugin/lirafusion/datasets/loading.py:954
Class
RandomFlip3D_radar
Flip the points & bbox. If the input dict contains the key "flip", then the flag will be used, otherwise it will be randomly decided by a rat
plugin/lirafusion/datasets/radar_points.py:297
Class
RandomScaleImageMultiViewImage
Random scale the image Args: scales
plugin/lirafusion/datasets/transform_3d.py:224
Class
ReducedDataBaseSampler
Class for sampling data from the ground truth database. Args: info_path (str): Path of groundtruth database info. data_root (st
plugin/lirafusion/datasets/loading.py:326
Class
fuse_lidar_radar
Fuse lidar and radar points.
plugin/lirafusion/datasets/loading.py:911
Class
voxel_fusion_layer
plugin/lirafusion/models/backbones/voxel_fusion_layer.py:16