highch

1 year ago · 97b6d9b02d
parent 1a6c0f2f83
commit 97b6d9b02d
1 changed files with 3 additions and 3 deletions
--- a/ultralytics/models/v8/yolov8-afpn.yaml
+++ b/ultralytics/models/v8/yolov8-afpn.yaml
@ -64,7 +64,7 @@ head:
  - [[14, 15, 16], 1, ASFF3, [256, 1]] # 18
  - [[14, 15, 16], 1, ASFF3, [512, 2]] # 19

-  - [17, 1, C2f, [128, False]]  # 20 (P3/8-small)
-  - [18, 1, C2f, [256, False]]  # 21 (P4/16-medium)
-  - [19, 1, C2f, [512, False]]  # 22 (P5/32-large)
+  - [17, 1, C2f, [256, False]]  # 20 (P3/8-small)
+  - [18, 1, C2f, [512, False]]  # 21 (P4/16-medium)
+  - [19, 1, C2f, [1024, False]]  # 22 (P5/32-large)
  - [[20, 21, 22], 1, Detect, [nc]]  # Detect(P3, P4, P5)