Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -108,16 +108,16 @@ inputs = [
|
|
108 |
value="NMS",
|
109 |
label="postprocess_type",
|
110 |
),
|
111 |
-
gr.
|
112 |
["IOU", "IOS"], type="value", default="IOU", label="postprocess_type"
|
113 |
),
|
114 |
-
gr.
|
115 |
-
gr.
|
116 |
]
|
117 |
|
118 |
outputs = [
|
119 |
-
gr.
|
120 |
-
gr.
|
121 |
]
|
122 |
|
123 |
title = "Small Object Detection with SAHI + YOLOv5"
|
|
|
108 |
value="NMS",
|
109 |
label="postprocess_type",
|
110 |
),
|
111 |
+
gr.Dropdown(
|
112 |
["IOU", "IOS"], type="value", default="IOU", label="postprocess_type"
|
113 |
),
|
114 |
+
gr.Number(default=0.5, label="postprocess_match_threshold"),
|
115 |
+
gr.Checkbox(default=True, label="postprocess_class_agnostic"),
|
116 |
]
|
117 |
|
118 |
outputs = [
|
119 |
+
gr.Image(type="pil", label="YOLOv5s"),
|
120 |
+
gr.Image(type="pil", label="YOLOv5s + SAHI"),
|
121 |
]
|
122 |
|
123 |
title = "Small Object Detection with SAHI + YOLOv5"
|