From 3f443b453946616a16d2a1bad309bfde300db8aa Mon Sep 17 00:00:00 2001
From: shiyu22 <shiyu.chen@zilliz.com>
Date: Mon, 11 Apr 2022 19:47:27 +0800
Subject: [PATCH] Update yolov5 boxes

Signed-off-by: shiyu22 <shiyu.chen@zilliz.com>
---
 yolov5.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/yolov5.py b/yolov5.py
index 9e927f9..477d9d6 100644
--- a/yolov5.py
+++ b/yolov5.py
@@ -14,7 +14,7 @@ class Yolov5(NNOperator):
         # Get object detection results with YOLOv5 model
         results = self._model(img)
 
-        boxes = [re[0:3] for re in results.xyxy[0]]
+        boxes = [re[0:4] for re in results.xyxy[0]]
         boxes = [list(map(int, box)) for box in boxes]
         classes = list(results.pandas().xyxy[0].name)
         scores = list(results.pandas().xyxy[0].confidence)