BhumikaMak commited on
Commit
4345023
·
1 Parent(s): 591dc67

Fix: version dependency

Browse files
Files changed (1) hide show
  1. app.py +2 -0
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())