Spaces:
Running
Running
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"
|