HongChuang/ultralytics-ascend
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 3

Clean up unused Silence module (#13674)

Browse source
This commit is contained in:
Laughing 2024-06-17 21:18:19 +08:00 • committed by GitHub
parent 6227d8f8a1
commit 605e7f4f52
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
4 changed files with 2 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

2
ultralytics/cfg/models/v9/yolov9e-seg.yaml
View file

@ -7,7 +7,7 @@ nc: 80 # number of classes
# gelan backbone # gelan backbone
backbone: backbone:
- [-1, 1, Silence, []] - [-1, 1, nn.Identity, []]
- [-1, 1, Conv, [64, 3, 2]] # 1-P1/2 - [-1, 1, Conv, [64, 3, 2]] # 1-P1/2
- [-1, 1, Conv, [128, 3, 2]] # 2-P2/4 - [-1, 1, Conv, [128, 3, 2]] # 2-P2/4
- [-1, 1, RepNCSPELAN4, [256, 128, 64, 2]] # 3 - [-1, 1, RepNCSPELAN4, [256, 128, 64, 2]] # 3

2
ultralytics/cfg/models/v9/yolov9e.yaml
View file

@ -7,7 +7,7 @@ nc: 80 # number of classes
# gelan backbone # gelan backbone
backbone: backbone:
- [-1, 1, Silence, []] - [-1, 1, nn.Identity, []]
- [-1, 1, Conv, [64, 3, 2]] # 1-P1/2 - [-1, 1, Conv, [64, 3, 2]] # 1-P1/2
- [-1, 1, Conv, [128, 3, 2]] # 2-P2/4 - [-1, 1, Conv, [128, 3, 2]] # 2-P2/4
- [-1, 1, RepNCSPELAN4, [256, 128, 64, 2]] # 3 - [-1, 1, RepNCSPELAN4, [256, 128, 64, 2]] # 3

2
ultralytics/nn/modules/__init__.py
View file

@ -45,7 +45,6 @@ from .block import (
RepC3, RepC3,
RepNCSPELAN4, RepNCSPELAN4,
ResNetLayer, ResNetLayer,
Silence,
) )
from .conv import ( from .conv import (
CBAM, CBAM,
@ -134,5 +133,4 @@ __all__ = (
"SPPELAN", "SPPELAN",
"CBFuse", "CBFuse",
"CBLinear", "CBLinear",
"Silence",
) )

1
ultralytics/nn/tasks.py
View file

@ -47,7 +47,6 @@ from ultralytics.nn.modules import (
ResNetLayer, ResNetLayer,
RTDETRDecoder, RTDETRDecoder,
Segment, Segment,
Silence, # noqa (equivalent to nn.Identity, unused in file but required to load YOLOv9 models)
WorldDetect, WorldDetect,
) )
from ultralytics.utils import DEFAULT_CFG_DICT, DEFAULT_CFG_KEYS, LOGGER, colorstr, emojis, yaml_load from ultralytics.utils import DEFAULT_CFG_DICT, DEFAULT_CFG_KEYS, LOGGER, colorstr, emojis, yaml_load