Add File

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,
+}