diff --git a/pcdet/models/backbones_3d/vfe/__init__.py b/pcdet/models/backbones_3d/vfe/__init__.py new file mode 100644 index 0000000..e666016 --- /dev/null +++ b/pcdet/models/backbones_3d/vfe/__init__.py @@ -0,0 +1,18 @@ +from .mean_vfe import MeanVFE +from .pillar_vfe import PillarVFE +from .dynamic_mean_vfe import DynamicMeanVFE +from .dynamic_pillar_vfe import DynamicPillarVFE, DynamicPillarVFESimple2D +from .dynamic_voxel_vfe import DynamicVoxelVFE +from .image_vfe import ImageVFE +from .vfe_template import VFETemplate + +__all__ = { + 'VFETemplate': VFETemplate, + 'MeanVFE': MeanVFE, + 'PillarVFE': PillarVFE, + 'ImageVFE': ImageVFE, + 'DynMeanVFE': DynamicMeanVFE, + 'DynPillarVFE': DynamicPillarVFE, + 'DynamicPillarVFESimple2D': DynamicPillarVFESimple2D, + 'DynamicVoxelVFE': DynamicVoxelVFE, +}