From 35fe03146187cd8b1c09dd3e72ae678bb9ec5b86 Mon Sep 17 00:00:00 2001 From: Yukun Xia Date: Mon, 30 Aug 2021 09:46:33 -0400 Subject: [PATCH] Fix: add P2 layer 21 to yolov5-p2.yaml `Detect()` inputs (#4608) Layer 21 includes the information of xsmall objects --- models/hub/yolov5-p2.yaml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/models/hub/yolov5-p2.yaml b/models/hub/yolov5-p2.yaml index 44d8da55dafb..759e9f92fb29 100644 --- a/models/hub/yolov5-p2.yaml +++ b/models/hub/yolov5-p2.yaml @@ -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) ]