linhcuem commited on
Commit
76fc1cc
·
1 Parent(s): 7ad4879

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -50,7 +50,7 @@ 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('hop_dln')
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
 
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('hop_dln')]
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