add transfusion head

Signed-off-by: wep21 <daisuke.nishimatsu1021@gmail.com>
open-mmlab · Jun 6, 2023 · bb0a95e · bb0a95e
1 parent 4b2e8c8
commit bb0a95e
Show file tree

Hide file tree

Showing 2 changed files with 847 additions and 1 deletion.
diff --git a/projects/TransFusion/transfusion/__init__.py b/projects/TransFusion/transfusion/__init__.py
@@ -1,7 +1,7 @@
-from ...BEVFusion.bevfusion.transfusion_head import TransFusionHead
 from ...BEVFusion.bevfusion.utils import (HungarianAssigner3D,
                                           TransFusionBBoxCoder)
 from .transfusion import TransFusion
+from .transfusion_head import TransFusionHead
 
 __all__ = [
     'HungarianAssigner3D',