Browse Source
Update yolov5 boxes
Signed-off-by: shiyu22 <shiyu.chen@zilliz.com>
main
1 changed files with
1 additions and
1 deletions
-
yolov5.py
|
|
@ -14,7 +14,7 @@ class Yolov5(NNOperator): |
|
|
# Get object detection results with YOLOv5 model |
|
|
# Get object detection results with YOLOv5 model |
|
|
results = self._model(img) |
|
|
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] |
|
|
boxes = [list(map(int, box)) for box in boxes] |
|
|
classes = list(results.pandas().xyxy[0].name) |
|
|
classes = list(results.pandas().xyxy[0].name) |
|
|
scores = list(results.pandas().xyxy[0].confidence) |
|
|
scores = list(results.pandas().xyxy[0].confidence) |
|
|
|
