Xu Ma commited on
Commit
7e9a1f7
1 Parent(s): 29a4916

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +12 -2
app.py CHANGED
@@ -117,7 +117,6 @@ def export_json(results, model, img_size):
117
  ]
118
 
119
 
120
- # YOLOv5图片检测函数
121
  def yolo_det(img, device, model_name, inference_size, conf, iou, label_opt, model_cls):
122
 
123
  global model, model_name_tmp, device_tmp
@@ -190,7 +189,18 @@ def main(args):
190
  model_cls_name = yaml_csv(cls_name, "model_cls_name")
191
 
192
  # -------------------输入组件-------------------
193
- inputs_img = gr.inputs.Image(type="pil", label="原始图片")
 
 
 
 
 
 
 
 
 
 
 
194
  device = gr.inputs.Dropdown(
195
  choices=["cpu"], default=device, type="value", label="设备"
196
  )
 
117
  ]
118
 
119
 
 
120
  def yolo_det(img, device, model_name, inference_size, conf, iou, label_opt, model_cls):
121
 
122
  global model, model_name_tmp, device_tmp
 
189
  model_cls_name = yaml_csv(cls_name, "model_cls_name")
190
 
191
  # -------------------输入组件-------------------
192
+ inputs_img = gr.inputs.Image(type="pil", label="Input Image")
193
+ experiment_id = gr.inputs.Radio(
194
+ choices=[
195
+ "add [1,1,1,1,1] paths",
196
+ "add [1,1,1,1,1,1,1,1] paths",
197
+ "add [1,2,4,8,16,32, ...] total 128 paths",
198
+ "add [1,2,4,8,16,32, ...] total 256 paths"], type="value", default="add [1,1,1,1,1] paths", label="Path Adding Scheduler"
199
+ )
200
+
201
+
202
+
203
+
204
  device = gr.inputs.Dropdown(
205
  choices=["cpu"], default=device, type="value", label="设备"
206
  )