Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -50,8 +50,8 @@ def yolov8_img_inference(
|
|
50 |
# get the model names list
|
51 |
names = model.names
|
52 |
# get the 'obj' class id
|
53 |
-
obj_id = list(names)[list(names.values()).index('
|
54 |
-
# ,'hop_jn','hop_vtg','hop_ytv','lo_kids', 'lo_ytv','loc_dln','loc_jn','loc_kids','loc_ytv')]
|
55 |
# obj_id = list(names)[list(names.values()).index([0])]
|
56 |
# count 'car' objects in the results
|
57 |
# count_result = results[0].boxes.cls[0].item()
|
|
|
50 |
# get the model names list
|
51 |
names = model.names
|
52 |
# get the 'obj' class id
|
53 |
+
obj_id = list(names)[list(names.values()).index('lo_ytv')]
|
54 |
+
# ('hop_dln','hop_jn','hop_vtg','hop_ytv','lo_kids', 'lo_ytv','loc_dln','loc_jn','loc_kids','loc_ytv')]
|
55 |
# obj_id = list(names)[list(names.values()).index([0])]
|
56 |
# count 'car' objects in the results
|
57 |
# count_result = results[0].boxes.cls[0].item()
|