Spaces:
Runtime error
Runtime error
Zengyf-CVer
commited on
Commit
•
65a0495
1
Parent(s):
12da3ca
v03 update
Browse files
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])
|