Update yolov5 boxes

Signed-off-by: shiyu22 <shiyu.chen@zilliz.com>
3 years ago · 3f443b4539
1 changed files with 1 additions and 1 deletions
--- 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)