From 1db6554cf2f85b6a364916a51716a836fb4d1a78 Mon Sep 17 00:00:00 2001
From: Yukun Xia <yukunx@andrew.cmu.edu>
Date: Mon, 30 Aug 2021 09:41:20 -0400
Subject: [PATCH] fix: add layer 21 to the final detection list

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)
   ]