Zengyf-CVer commited on
Commit
455cf48
1 Parent(s): e0f8c82

v04 add csv excel

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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