Zengyf-CVer commited on
Commit
65a0495
1 Parent(s): 12da3ca

v03 update

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -219,6 +219,7 @@ def yolo_det(img, device, model_name, inference_size, conf, iou, max_num, model_
219
  id = int(i) # instance ID
220
  obj_cls_index = int(result[i][5]) # category index
221
  obj_cls = model_cls_name_cp[obj_cls_index] # category
 
222
 
223
  # ------------ border coordinates ------------
224
  x0 = float(result[i][:4].tolist()[0])
 
219
  id = int(i) # instance ID
220
  obj_cls_index = int(result[i][5]) # category index
221
  obj_cls = model_cls_name_cp[obj_cls_index] # category
222
+ cls_det_stat.append(obj_cls)
223
 
224
  # ------------ border coordinates ------------
225
  x0 = float(result[i][:4].tolist()[0])