Zengyf-CVer commited on
Commit
323ca06
1 Parent(s): 0938b66

v04 update

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -326,7 +326,7 @@ def yolo_det_img(img, device, model_name, infer_size, conf, iou, max_num, model_
326
  for k, v in clsDet_dict.items():
327
  clsRatio_dict[k] = v / clsDet_dict_sum
328
 
329
- return det_img, objSize_dict, clsRatio_dict, det_json, report, dataframe, img_crops
330
 
331
 
332
  # YOLOv5 video detection function
@@ -529,7 +529,7 @@ def main(args):
529
  outputs_video = gr.Video(format='mp4', label="Detection video")
530
 
531
  # output parameters
532
- outputs_img_list = [outputs_img, outputs_objSize, outputs_clsSize, outputs_json, outputs_pdf, outputs_df, outputs_crops]
533
  outputs_video_list = [outputs_video]
534
 
535
  # title
326
  for k, v in clsDet_dict.items():
327
  clsRatio_dict[k] = v / clsDet_dict_sum
328
 
329
+ return det_img, img_crops, objSize_dict, clsRatio_dict, dataframe, det_json, report
330
 
331
 
332
  # YOLOv5 video detection function
529
  outputs_video = gr.Video(format='mp4', label="Detection video")
530
 
531
  # output parameters
532
+ outputs_img_list = [outputs_img, outputs_crops, outputs_objSize, outputs_clsSize, outputs_df, outputs_json, outputs_pdf]
533
  outputs_video_list = [outputs_video]
534
 
535
  # title