Spaces:
Runtime error
Runtime error
Xu Ma
commited on
Commit
•
7e9a1f7
1
Parent(s):
29a4916
Update app.py
Browse files
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 |
)
|