diff --git a/ultralytics/cfg/models/v8/yolov8-2.yaml b/ultralytics/cfg/models/v8/yolov8-c2fn-2222.yaml similarity index 84% rename from ultralytics/cfg/models/v8/yolov8-2.yaml rename to ultralytics/cfg/models/v8/yolov8-c2fn-2222.yaml index 25011d6e58..bbdb09db14 100644 --- a/ultralytics/cfg/models/v8/yolov8-2.yaml +++ b/ultralytics/cfg/models/v8/yolov8-c2fn-2222.yaml @@ -16,31 +16,31 @@ backbone: # [from, repeats, module, args] - [-1, 1, Conv, [64, 3, 2]] # 0-P1/2 - [-1, 1, Conv, [128, 3, 2]] # 1-P2/4 - - [-1, 3, C2fn, [128, True]] + - [-1, 2, C2fn, [128, True]] - [-1, 1, Conv, [256, 3, 2]] # 3-P3/8 - - [-1, 6, C2fn, [256, True]] + - [-1, 2, C2fn, [256, True]] - [-1, 1, Conv, [512, 3, 2]] # 5-P4/16 - - [-1, 6, C2fn, [512, True]] + - [-1, 2, C2fn, [512, True]] - [-1, 1, Conv, [1024, 3, 2]] # 7-P5/32 - - [-1, 3, C2fn, [1024, True]] + - [-1, 2, C2fn, [1024, True]] - [-1, 1, SPPF, [1024, 5]] # 9 # YOLOv8.0n head head: - [-1, 1, nn.Upsample, [None, 2, "nearest"]] - [[-1, 6], 1, Concat, [1]] # cat backbone P4 - - [-1, 3, C2fn, [512]] # 12 + - [-1, 2, C2fn, [512]] # 12 - [-1, 1, nn.Upsample, [None, 2, "nearest"]] - [[-1, 4], 1, Concat, [1]] # cat backbone P3 - - [-1, 3, C2fn, [256]] # 15 (P3/8-small) + - [-1, 2, C2fn, [256]] # 15 (P3/8-small) - [-1, 1, Conv, [256, 3, 2]] - [[-1, 12], 1, Concat, [1]] # cat head P4 - - [-1, 3, C2fn, [512]] # 18 (P4/16-medium) + - [-1, 2, C2fn, [512]] # 18 (P4/16-medium) - [-1, 1, Conv, [512, 3, 2]] - [[-1, 9], 1, Concat, [1]] # cat head P5 - - [-1, 3, C2fn, [1024]] # 21 (P5/32-large) + - [-1, 2, C2fn, [1024]] # 21 (P5/32-large) - [[15, 18, 21], 1, Detect, [nc]] # Detect(P3, P4, P5) diff --git a/ultralytics/nn/modules/block.py b/ultralytics/nn/modules/block.py index 90354021fd..0ee577da1b 100644 --- a/ultralytics/nn/modules/block.py +++ b/ultralytics/nn/modules/block.py @@ -250,7 +250,7 @@ class C2fn(nn.Module): self.c = int(c2 * e) # hidden channels self.cv1 = Conv(c1, 2 * self.c, 1, 1) self.cv2 = Conv((2 + n) * self.c, c2, 1) # optional act=FReLU(c2) - self.m = nn.ModuleList(C2f(self.c, self.c, 1, shortcut, g) for _ in range(n)) + self.m = nn.ModuleList(C2f(self.c, self.c, 2, shortcut, g) for _ in range(n)) def forward(self, x): """Forward pass through C2f layer."""