Spaces:
Runtime error
Runtime error
File size: 3,936 Bytes
884c49d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 |
import gradio as gr
from metaseg import SegAutoMaskGenerator
def image_app():
with gr.Blocks():
with gr.Row():
with gr.Column():
seg_automask_image_file = gr.Image(type="filepath").style(height=260)
with gr.Row():
with gr.Column():
seg_automask_image_model_type = gr.Dropdown(
choices=[
"vit_h",
"vit_l",
"vit_b",
],
value="vit_l",
label="Model Type",
)
seg_automask_image_points_per_side = gr.Slider(
minimum=0,
maximum=32,
step=2,
value=16,
label="Points per Side",
)
seg_automask_image_points_per_batch = gr.Slider(
minimum=0,
maximum=64,
step=2,
value=64,
label="Points per Batch",
)
seg_automask_image_predict = gr.Button(value="Generator")
with gr.Column():
output_image = gr.Image()
seg_automask_image_predict.click(
fn=SegAutoMaskGenerator().save_image,
inputs=[
seg_automask_image_file,
seg_automask_image_model_type,
seg_automask_image_points_per_side,
seg_automask_image_points_per_batch,
],
outputs=[output_image],
)
def video_app():
with gr.Blocks():
with gr.Row():
with gr.Column():
seg_automask_video_file = gr.Video().style(height=260)
with gr.Row():
with gr.Column():
seg_automask_video_model_type = gr.Dropdown(
choices=[
"vit_h",
"vit_l",
"vit_b",
],
value="vit_l",
label="Model Type",
)
seg_automask_video_points_per_side = gr.Slider(
minimum=0,
maximum=32,
step=2,
value=16,
label="Points per Side",
)
seg_automask_video_points_per_batch = gr.Slider(
minimum=0,
maximum=64,
step=2,
value=64,
label="Points per Batch",
)
seg_automask_video_predict = gr.Button(value="Generator")
with gr.Column():
output_video = gr.Video()
seg_automask_video_predict.click(
fn=SegAutoMaskGenerator().save_image,
inputs=[
seg_automask_video_file,
seg_automask_video_model_type,
seg_automask_video_points_per_side,
seg_automask_video_points_per_batch,
],
outputs=[output_video],
)
def metaseg_app():
app = gr.Blocks()
with app:
with gr.Row():
with gr.Column():
with gr.Tab("Image"):
image_app()
with gr.Tab("Video"):
video_app()
app.queue(concurrency_count=2)
app.launch(debug=True, enable_queue=True)
if __name__ == "__main__":
metaseg_app()
|