From 44f72f24e2c08a7492e0bc1d4b27fd2acf299a4c Mon Sep 17 00:00:00 2001
From: inter <inter@qq.com>
Date: Sun, 21 Sep 2025 20:19:02 +0800
Subject: [PATCH] Add File

---
 pcdet/models/roi_heads/__init__.py | 19 +++++++++++++++++++
 1 file changed, 19 insertions(+)
 create mode 100644 pcdet/models/roi_heads/__init__.py

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