diff --git a/pcdet/models/dense_heads/__init__.py b/pcdet/models/dense_heads/__init__.py
new file mode 100644
index 0000000..3bb33de
--- /dev/null
+++ b/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,
+}