exp-a
Laughing-q 10 months ago
parent e5224e5171
commit 21219fe443
  1. 16
      ultralytics/cfg/models/v8/yolov8-c2fn-2222.yaml
  2. 2
      ultralytics/nn/modules/block.py

@ -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)

@ -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."""

Loading…
Cancel
Save