Yukun Xia
commited on
Commit
•
35fe031
1
Parent(s):
e7d1842
Fix: add P2 layer 21 to yolov5-p2.yaml `Detect()` inputs (#4608)
Browse files
models/hub/yolov5-p2.yaml
CHANGED
|
@@ -50,5 +50,5 @@ head:
|
|
| 50 |
[[-1, 10], 1, Concat, [1]], # cat head P5
|
| 51 |
[-1, 3, C3, [1024, False]], # 30 (P5/32-large)
|
| 52 |
|
| 53 |
-
[[24, 27, 30], 1, Detect, [nc, anchors]], # Detect(P3, P4, P5)
|
| 54 |
]
|
|
|
|
| 50 |
[[-1, 10], 1, Concat, [1]], # cat head P5
|
| 51 |
[-1, 3, C3, [1024, False]], # 30 (P5/32-large)
|
| 52 |
|
| 53 |
+
[[21, 24, 27, 30], 1, Detect, [nc, anchors]], # Detect(P2, P3, P4, P5)
|
| 54 |
]
|