Revert change in init

Add heads by need only
HikariTJU · Oct 24, 2024 · 37872c2 · 37872c2
1 parent 6b954fe
commit 37872c2
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/mmdet/models/dense_heads/__init__.py b/mmdet/models/dense_heads/__init__.py
@@ -23,9 +23,7 @@
 from .retina_sepbn_head import RetinaSepBNHead
 from .rpn_head import RPNHead
 from .sabl_retina_head import SABLRetinaHead
-from .solo_head import SOLOHead
 from .ssd_head import SSDHead
-from .tood_head import TOODHead
 from .transformer_head import TransformerHead
 from .vfnet_head import VFNetHead
 from .yolact_head import YOLACTHead, YOLACTProtonet, YOLACTSegmHead
@@ -47,5 +45,5 @@
     'YOLACTSegmHead', 'YOLACTProtonet', 'YOLOV3Head', 'PAAHead',
     'SABLRetinaHead', 'CentripetalHead', 'VFNetHead', 'TransformerHead',
     'StageCascadeRPNHead', 'CascadeRPNHead', 'EmbeddingRPNHead', 'LDHead',
-    'GFocalHead', 'IMv2Head', 'SOLOHead', 'TOODHead'
+    'GFocalHead', 'IMv2Head'
 ]