Spaces:
Sleeping
Sleeping
Commit
·
4345023
1
Parent(s):
591dc67
Fix: version dependency
Browse files
app.py
CHANGED
@@ -33,7 +33,9 @@ def parse_detections(results, yolo_version):
|
|
33 |
elif yolo_version == "yolov8":
|
34 |
# For YOLOv8
|
35 |
for result in results:
|
|
|
36 |
for box in result.boxes:
|
|
|
37 |
xmin, ymin, xmax, ymax = box[0].xyxy.tolist()
|
38 |
confidence = box.conf.item()
|
39 |
class_id = int(box.cls.item())
|
|
|
33 |
elif yolo_version == "yolov8":
|
34 |
# For YOLOv8
|
35 |
for result in results:
|
36 |
+
print('resukt', result)
|
37 |
for box in result.boxes:
|
38 |
+
print('box', box)
|
39 |
xmin, ymin, xmax, ymax = box[0].xyxy.tolist()
|
40 |
confidence = box.conf.item()
|
41 |
class_id = int(box.cls.item())
|