add transfusion head

Signed-off-by: wep21 <[email protected]>
open-mmlab · Aug 18, 2023 · 6b2b6ba · 6b2b6ba
1 parent c74b50c
commit 6b2b6ba
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',