Ankle/yolov5_birdproject
1
0 Fork 0
Code Issues Pull Requests Actions 6 Packages Projects Releases Wiki Activity

Fix: add P2 layer 21 to yolov5-p2.yaml Detect() inputs (#4608)

Browse Source 
Layer 21 includes the information of xsmall objects
This commit is contained in:
Yukun Xia authored and GitHub committed 2021-08-30 15:46:33 +02:00
1 parent e7d1842a98
commit 35fe031461
1 file changed
+1 -1
Download Patch File Download Diff File Expand all files Collapse all files
models/hub/yolov5-p2.yaml
+1 -1
View File
@@ -50,5 +50,5 @@ head:
[[-1, 10], 1, Concat, [1]], # cat head P5 [[-1, 10], 1, Concat, [1]], # cat head P5
[-1, 3, C3, [1024, False]], # 30 (P5/32-large) [-1, 3, C3, [1024, False]], # 30 (P5/32-large)
[[24, 27, 30], 1, Detect, [nc, anchors]], # Detect(P3, P4, P5) [[21, 24, 27, 30], 1, Detect, [nc, anchors]], # Detect(P2, P3, P4, P5)
] ]