diff --git a/pcdet/models/roi_heads/__init__.py b/pcdet/models/roi_heads/__init__.py
new file mode 100644
index 0000000..693cec4
--- /dev/null
+++ b/pcdet/models/roi_heads/__init__.py
@@ -0,0 +1,19 @@
+from .partA2_head import PartA2FCHead
+from .pointrcnn_head import PointRCNNHead
+from .pvrcnn_head import PVRCNNHead
+from .second_head import SECONDHead
+from .voxelrcnn_head import VoxelRCNNHead
+from .roi_head_template import RoIHeadTemplate
+from .mppnet_head import MPPNetHead
+from .mppnet_memory_bank_e2e import MPPNetHeadE2E
+
+__all__ = {
+    'RoIHeadTemplate': RoIHeadTemplate,
+    'PartA2FCHead': PartA2FCHead,
+    'PVRCNNHead': PVRCNNHead,
+    'SECONDHead': SECONDHead,
+    'PointRCNNHead': PointRCNNHead,
+    'VoxelRCNNHead': VoxelRCNNHead,
+    'MPPNetHead': MPPNetHead,
+    'MPPNetHeadE2E': MPPNetHeadE2E,
+}