Clean up unused Silence module (ultralytics#13674)

Burhan-Q · Jun 17, 2024 · 605e7f4 · 605e7f4
1 parent 6227d8f
commit 605e7f4
Show file tree

Hide file tree

Showing 4 changed files with 2 additions and 5 deletions.
diff --git a/ultralytics/cfg/models/v9/yolov9e-seg.yaml b/ultralytics/cfg/models/v9/yolov9e-seg.yaml
@@ -7,7 +7,7 @@ nc: 80 # number of classes
 
 # gelan backbone
 backbone:
-  - [-1, 1, Silence, []]
+  - [-1, 1, nn.Identity, []]
   - [-1, 1, Conv, [64, 3, 2]] # 1-P1/2
   - [-1, 1, Conv, [128, 3, 2]] # 2-P2/4
   - [-1, 1, RepNCSPELAN4, [256, 128, 64, 2]] # 3

diff --git a/ultralytics/cfg/models/v9/yolov9e.yaml b/ultralytics/cfg/models/v9/yolov9e.yaml
@@ -7,7 +7,7 @@ nc: 80 # number of classes
 
 # gelan backbone
 backbone:
-  - [-1, 1, Silence, []]
+  - [-1, 1, nn.Identity, []]
   - [-1, 1, Conv, [64, 3, 2]] # 1-P1/2
   - [-1, 1, Conv, [128, 3, 2]] # 2-P2/4
   - [-1, 1, RepNCSPELAN4, [256, 128, 64, 2]] # 3

diff --git a/ultralytics/nn/modules/__init__.py b/ultralytics/nn/modules/__init__.py
@@ -45,7 +45,6 @@
     RepC3,
     RepNCSPELAN4,
     ResNetLayer,
-    Silence,
 )
 from .conv import (
     CBAM,
@@ -134,5 +133,4 @@
     "SPPELAN",
     "CBFuse",
     "CBLinear",
-    "Silence",
 )
diff --git a/ultralytics/nn/tasks.py b/ultralytics/nn/tasks.py
@@ -47,7 +47,6 @@
     ResNetLayer,
     RTDETRDecoder,
     Segment,
-    Silence,  # noqa (equivalent to nn.Identity, unused in file but required to load YOLOv9 models)
     WorldDetect,
 )
 from ultralytics.utils import DEFAULT_CFG_DICT, DEFAULT_CFG_KEYS, LOGGER, colorstr, emojis, yaml_load