Spaces:

linhcuem
/

cham_diem_yolov8

Sleeping

linhcuem commited on Oct 24, 2023

Commit

680ea31

1 Parent(s): b3be888

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -55,7 +55,7 @@ def yolov8_img_inference(
     # obj_id = list(names)[list(names.values()).index([0])]
     # count 'car' objects in the results
     # count_result = results[0].boxes.cls[0].item()
-    count_result = len(results.boxes.data)
     object_counts = {x: 0 for x in names}
     for r in results:

     # obj_id = list(names)[list(names.values()).index([0])]
     # count 'car' objects in the results
     # count_result = results[0].boxes.cls[0].item()
+    count_result = results.boxes.data
     object_counts = {x: 0 for x in names}
     for r in results: