Spaces:
Runtime error
Runtime error
Zengyf-CVer
commited on
Commit
•
323ca06
1
Parent(s):
0938b66
v04 update
Browse files
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
|
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
|
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
|