diff --git a/pcdet/models/backbones_3d/vfe/image_vfe_modules/ffn/__init__.py b/pcdet/models/backbones_3d/vfe/image_vfe_modules/ffn/__init__.py
new file mode 100644
index 0000000..576d5eb
--- /dev/null
+++ b/pcdet/models/backbones_3d/vfe/image_vfe_modules/ffn/__init__.py
@@ -0,0 +1,5 @@
+from .depth_ffn import DepthFFN
+
+__all__ = {
+    'DepthFFN': DepthFFN
+}