Spaces:
Runtime error
Runtime error
update
Browse files
app.py
CHANGED
|
@@ -120,61 +120,31 @@ def sahi_app():
|
|
| 120 |
with gr.Row():
|
| 121 |
with gr.Column():
|
| 122 |
sahi_image_file = gr.Image(type="filepath").style(height=260)
|
| 123 |
-
|
| 124 |
with gr.Row():
|
| 125 |
with gr.Column():
|
| 126 |
sahi_autoseg_model_type = gr.Dropdown(
|
| 127 |
-
|
| 128 |
-
|
| 129 |
-
|
| 130 |
-
|
| 131 |
-
|
| 132 |
-
|
| 133 |
-
|
| 134 |
-
|
| 135 |
-
sahi_model_type = gr.Dropdown(
|
| 136 |
-
choices=[
|
| 137 |
-
"yolov5",
|
| 138 |
-
"yolov8",
|
| 139 |
-
],
|
| 140 |
-
value="yolov5",
|
| 141 |
-
label="Detector Model Type",
|
| 142 |
-
)
|
| 143 |
-
|
| 144 |
-
sahi_model_path = gr.Dropdown(
|
| 145 |
-
choices=[
|
| 146 |
-
"yolov5m",
|
| 147 |
-
"yolov5l",
|
| 148 |
-
"yolov5m6",
|
| 149 |
-
"yolov5l6",
|
| 150 |
-
],
|
| 151 |
-
value="yolov5m",
|
| 152 |
-
label="Detector Model Path",
|
| 153 |
-
)
|
| 154 |
-
sahi_conf_th = gr.Slider(
|
| 155 |
-
minimum=0,
|
| 156 |
-
maximum=1,
|
| 157 |
-
step=0.1,
|
| 158 |
-
value=0.2,
|
| 159 |
-
label="Confidence Threshold",
|
| 160 |
-
)
|
| 161 |
-
|
| 162 |
sahi_image_size = gr.Slider(
|
| 163 |
minimum=0,
|
| 164 |
maximum=1600,
|
| 165 |
step=32,
|
| 166 |
value=640,
|
| 167 |
label="Image Size",
|
| 168 |
-
)
|
| 169 |
-
|
| 170 |
-
sahi_slice_height = gr.Slider(
|
| 171 |
minimum=0,
|
| 172 |
-
maximum=
|
| 173 |
-
step=
|
| 174 |
-
value=
|
| 175 |
-
label="
|
| 176 |
-
)
|
| 177 |
-
|
| 178 |
sahi_slice_width = gr.Slider(
|
| 179 |
minimum=0,
|
| 180 |
maximum=640,
|
|
@@ -190,14 +160,42 @@ def sahi_app():
|
|
| 190 |
value=0.2,
|
| 191 |
label="Overlap Height",
|
| 192 |
)
|
| 193 |
-
|
| 194 |
-
|
| 195 |
-
|
| 196 |
-
|
| 197 |
-
|
| 198 |
-
|
| 199 |
-
|
| 200 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 201 |
|
| 202 |
|
| 203 |
sahi_image_predict = gr.Button(value="Generator")
|
|
@@ -226,7 +224,7 @@ def sahi_app():
|
|
| 226 |
def metaseg_app():
|
| 227 |
app = gr.Blocks()
|
| 228 |
with app:
|
| 229 |
-
gr.Markdown("# **<h2 align='center'>Segment Anything + Video +
|
| 230 |
gr.Markdown(
|
| 231 |
"""
|
| 232 |
<h5 style='text-align: center'>
|
|
|
|
| 120 |
with gr.Row():
|
| 121 |
with gr.Column():
|
| 122 |
sahi_image_file = gr.Image(type="filepath").style(height=260)
|
|
|
|
| 123 |
with gr.Row():
|
| 124 |
with gr.Column():
|
| 125 |
sahi_autoseg_model_type = gr.Dropdown(
|
| 126 |
+
choices=[
|
| 127 |
+
"vit_h",
|
| 128 |
+
"vit_l",
|
| 129 |
+
"vit_b",
|
| 130 |
+
],
|
| 131 |
+
value="vit_l",
|
| 132 |
+
label="Sam Model Type",
|
| 133 |
+
)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 134 |
sahi_image_size = gr.Slider(
|
| 135 |
minimum=0,
|
| 136 |
maximum=1600,
|
| 137 |
step=32,
|
| 138 |
value=640,
|
| 139 |
label="Image Size",
|
| 140 |
+
)
|
| 141 |
+
sahi_conf_th = gr.Slider(
|
|
|
|
| 142 |
minimum=0,
|
| 143 |
+
maximum=1,
|
| 144 |
+
step=0.1,
|
| 145 |
+
value=0.2,
|
| 146 |
+
label="Confidence Threshold",
|
| 147 |
+
)
|
|
|
|
| 148 |
sahi_slice_width = gr.Slider(
|
| 149 |
minimum=0,
|
| 150 |
maximum=640,
|
|
|
|
| 160 |
value=0.2,
|
| 161 |
label="Overlap Height",
|
| 162 |
)
|
| 163 |
+
with gr.Row():
|
| 164 |
+
with gr.Column():
|
| 165 |
+
sahi_model_type = gr.Dropdown(
|
| 166 |
+
choices=[
|
| 167 |
+
"yolov5",
|
| 168 |
+
"yolov8",
|
| 169 |
+
],
|
| 170 |
+
value="yolov5",
|
| 171 |
+
label="Detector Model Type",
|
| 172 |
+
)
|
| 173 |
+
|
| 174 |
+
sahi_model_path = gr.Dropdown(
|
| 175 |
+
choices=[
|
| 176 |
+
"yolov5m",
|
| 177 |
+
"yolov5l",
|
| 178 |
+
"yolov5m6",
|
| 179 |
+
"yolov5l6",
|
| 180 |
+
],
|
| 181 |
+
value="yolov5m",
|
| 182 |
+
label="Detector Model Path",
|
| 183 |
+
)
|
| 184 |
+
sahi_slice_height = gr.Slider(
|
| 185 |
+
minimum=0,
|
| 186 |
+
maximum=640,
|
| 187 |
+
step=32,
|
| 188 |
+
value=256,
|
| 189 |
+
label="Slice Height",
|
| 190 |
+
)
|
| 191 |
+
|
| 192 |
+
sahi_overlap_width = gr.Slider(
|
| 193 |
+
minimum=0,
|
| 194 |
+
maximum=1,
|
| 195 |
+
step=0.1,
|
| 196 |
+
value=0.2,
|
| 197 |
+
label="Overlap Width",
|
| 198 |
+
)
|
| 199 |
|
| 200 |
|
| 201 |
sahi_image_predict = gr.Button(value="Generator")
|
|
|
|
| 224 |
def metaseg_app():
|
| 225 |
app = gr.Blocks()
|
| 226 |
with app:
|
| 227 |
+
gr.Markdown("# **<h2 align='center'>Segment Anything + Video + SAHI</h2>**")
|
| 228 |
gr.Markdown(
|
| 229 |
"""
|
| 230 |
<h5 style='text-align: center'>
|