Zengyf-CVer commited on
Commit
448468b
1 Parent(s): 3fb23cc

v03 update

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -271,7 +271,7 @@ def yolo_det(img, device, model_name, inference_size, conf, iou, max_num, model_
271
 
272
  objSize_dict = {obj_style[i]: [s_obj, m_obj, l_obj][i] / sml_obj_total for i in range(3)}
273
 
274
- return det_img, det_json, report, dataframe, objSize_dict
275
 
276
 
277
  def main(args):
@@ -337,7 +337,7 @@ def main(args):
337
  outputs_df = gr.outputs.Dataframe(max_rows=5, overflow_row_behaviour="paginate", type="pandas", label="List of detection information")
338
  outputs_objSize = gr.outputs.Label(label="Object size ratio statistics")
339
 
340
- outputs = [outputs_img, outputs_json, outputs_pdf, outputs_df, outputs_objSize]
341
 
342
  # title
343
  title = "Gradio YOLOv5 Det v0.3"
 
271
 
272
  objSize_dict = {obj_style[i]: [s_obj, m_obj, l_obj][i] / sml_obj_total for i in range(3)}
273
 
274
+ return det_img, objSize_dict, det_json, report, dataframe
275
 
276
 
277
  def main(args):
 
337
  outputs_df = gr.outputs.Dataframe(max_rows=5, overflow_row_behaviour="paginate", type="pandas", label="List of detection information")
338
  outputs_objSize = gr.outputs.Label(label="Object size ratio statistics")
339
 
340
+ outputs = [outputs_img, outputs_objSize, outputs_json, outputs_pdf, outputs_df]
341
 
342
  # title
343
  title = "Gradio YOLOv5 Det v0.3"