Spaces:
Build error
Build error
Change layout a bit
Browse files
app.py
CHANGED
|
@@ -255,15 +255,18 @@ This is an unofficial demo for [https://github.com/ViTAE-Transformer/ViTPose](ht
|
|
| 255 |
detect_button = gr.Button(value='Detect')
|
| 256 |
det_preds = gr.Variable()
|
| 257 |
with gr.Column():
|
| 258 |
-
|
| 259 |
-
|
| 260 |
-
|
| 261 |
-
|
| 262 |
-
|
| 263 |
-
|
| 264 |
-
|
| 265 |
-
|
| 266 |
-
|
|
|
|
|
|
|
|
|
|
| 267 |
|
| 268 |
with gr.Row():
|
| 269 |
paths = sorted(pathlib.Path('images').rglob('*.jpg'))
|
|
@@ -290,24 +293,30 @@ This is an unofficial demo for [https://github.com/ViTAE-Transformer/ViTPose](ht
|
|
| 290 |
predict_button = gr.Button(value='Predict')
|
| 291 |
pose_preds = gr.Variable()
|
| 292 |
with gr.Column():
|
| 293 |
-
|
| 294 |
-
|
| 295 |
-
|
| 296 |
-
|
| 297 |
-
|
| 298 |
-
|
| 299 |
-
|
| 300 |
-
|
| 301 |
-
|
| 302 |
-
|
| 303 |
-
|
| 304 |
-
|
| 305 |
-
vis_line_thickness = gr.Slider(1,
|
| 306 |
10,
|
| 307 |
step=1,
|
| 308 |
-
value=
|
| 309 |
-
label='
|
| 310 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 311 |
|
| 312 |
gr.Markdown(
|
| 313 |
'<center><img src="https://visitor-badge.glitch.me/badge?page_id=hysts.vitpose" alt="visitor badge"/></center>'
|
|
|
|
| 255 |
detect_button = gr.Button(value='Detect')
|
| 256 |
det_preds = gr.Variable()
|
| 257 |
with gr.Column():
|
| 258 |
+
with gr.Row():
|
| 259 |
+
detection_visualization = gr.Image(
|
| 260 |
+
label='Detection Result', type='numpy')
|
| 261 |
+
with gr.Row():
|
| 262 |
+
vis_det_score_threshold = gr.Slider(
|
| 263 |
+
0,
|
| 264 |
+
1,
|
| 265 |
+
step=0.05,
|
| 266 |
+
value=0.5,
|
| 267 |
+
label='Visualization Score Threshold')
|
| 268 |
+
with gr.Row():
|
| 269 |
+
redraw_det_button = gr.Button(value='Redraw')
|
| 270 |
|
| 271 |
with gr.Row():
|
| 272 |
paths = sorted(pathlib.Path('images').rglob('*.jpg'))
|
|
|
|
| 293 |
predict_button = gr.Button(value='Predict')
|
| 294 |
pose_preds = gr.Variable()
|
| 295 |
with gr.Column():
|
| 296 |
+
with gr.Row():
|
| 297 |
+
pose_visualization = gr.Image(label='Result',
|
| 298 |
+
type='numpy')
|
| 299 |
+
with gr.Row():
|
| 300 |
+
vis_kpt_score_threshold = gr.Slider(
|
| 301 |
+
0,
|
| 302 |
+
1,
|
| 303 |
+
step=0.05,
|
| 304 |
+
value=0.3,
|
| 305 |
+
label='Visualization Score Threshold')
|
| 306 |
+
with gr.Row():
|
| 307 |
+
vis_dot_radius = gr.Slider(1,
|
|
|
|
| 308 |
10,
|
| 309 |
step=1,
|
| 310 |
+
value=4,
|
| 311 |
+
label='Dot Radius')
|
| 312 |
+
with gr.Row():
|
| 313 |
+
vis_line_thickness = gr.Slider(1,
|
| 314 |
+
10,
|
| 315 |
+
step=1,
|
| 316 |
+
value=2,
|
| 317 |
+
label='Line Thickness')
|
| 318 |
+
with gr.Row():
|
| 319 |
+
redraw_pose_button = gr.Button(value='Redraw')
|
| 320 |
|
| 321 |
gr.Markdown(
|
| 322 |
'<center><img src="https://visitor-badge.glitch.me/badge?page_id=hysts.vitpose" alt="visitor badge"/></center>'
|