Skip to content

Commit

Permalink
Fix: add P2 layer 21 to yolov5-p2.yaml Detect() inputs (ultralytics…
Browse files Browse the repository at this point in the history
…#4608)

Layer 21 includes the information of xsmall objects
  • Loading branch information
YukunXia authored Aug 30, 2021
1 parent 5ea994d commit 864032d
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion models/hub/yolov5-p2.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -50,5 +50,5 @@ head:
[[-1, 10], 1, Concat, [1]], # cat head P5
[-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)
]

0 comments on commit 864032d

Please sign in to comment.