Spaces:

linhcuem
/

cham_diem_yolov8

Sleeping

linhcuem commited on Oct 20, 2023

Commit

9cb5733

1 Parent(s): 8e53412

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -50,7 +50,7 @@ def yolov8_img_inference(
     # get the model names list
     names = model.names
     # get the 'car' class id
-    obj_id = list(names)[list(names.values()).index('hop_dln','hop_jn','hop_vtg','hop_ytv','lo_kids', 'lo_ytv','loc_dln','loc_jn','loc_kids','loc_ytv')]
     # obj_id = list(names)[list(names.values()).index([0])]
     # count 'car' objects in the results
     # count_result = results[0].boxes.cls[0].item()

     # get the model names list
     names = model.names
     # get the 'car' class id
+    # obj_id = list(names)[list(names.values()).index('hop_dln','hop_jn','hop_vtg','hop_ytv','lo_kids', 'lo_ytv','loc_dln','loc_jn','loc_kids','loc_ytv')]
     # obj_id = list(names)[list(names.values()).index([0])]
     # count 'car' objects in the results
     # count_result = results[0].boxes.cls[0].item()