Add File

pcdet/models/dense_heads/__init__.py

@@ -0,0 +1,21 @@
+from .anchor_head_multi import AnchorHeadMulti
+from .anchor_head_single import AnchorHeadSingle
+from .anchor_head_template import AnchorHeadTemplate
+from .point_head_box import PointHeadBox
+from .point_head_simple import PointHeadSimple
+from .point_intra_part_head import PointIntraPartOffsetHead
+from .center_head import CenterHead
+from .voxelnext_head import VoxelNeXtHead
+from .transfusion_head import TransFusionHead
+
+__all__ = {
+    'AnchorHeadTemplate': AnchorHeadTemplate,
+    'AnchorHeadSingle': AnchorHeadSingle,
+    'PointIntraPartOffsetHead': PointIntraPartOffsetHead,
+    'PointHeadSimple': PointHeadSimple,
+    'PointHeadBox': PointHeadBox,
+    'AnchorHeadMulti': AnchorHeadMulti,
+    'CenterHead': CenterHead,
+    'VoxelNeXtHead': VoxelNeXtHead,
+    'TransFusionHead': TransFusionHead,
+}