Spaces:
Runtime error
Runtime error
Zengyf-CVer
commited on
Commit
·
455cf48
1
Parent(s):
e0f8c82
v04 add csv excel
Browse files
app.py
CHANGED
@@ -346,7 +346,7 @@ def yolo_det_img(img, device, model_name, infer_size, conf, iou, max_num, model_
|
|
346 |
for k, v in clsDet_dict.items():
|
347 |
clsRatio_dict[k] = v / clsDet_dict_sum
|
348 |
|
349 |
-
return det_img, img_crops, objSize_dict, clsRatio_dict, dataframe, det_json, report
|
350 |
|
351 |
|
352 |
# YOLOv5 video detection function
|
|
|
346 |
for k, v in clsDet_dict.items():
|
347 |
clsRatio_dict[k] = v / clsDet_dict_sum
|
348 |
|
349 |
+
return det_img, img_crops, objSize_dict, clsRatio_dict, dataframe, det_json, report, det_csv, det_excel
|
350 |
|
351 |
|
352 |
# YOLOv5 video detection function
|