Tobias Cornille commited on
Commit
7c85c0b
1 Parent(s): 549f527

Fix lay-out

Browse files
Files changed (1) hide show
  1. app.py +56 -55
app.py CHANGED
@@ -393,63 +393,64 @@ if __name__ == "__main__":
393
 
394
  block = gr.Blocks(title="Panoptic Segment Anything").queue()
395
  with block:
396
- title = gr.Markdown(
397
- "# [Panoptic Segment Anything](https://github.com/segments-ai/panoptic-segment-anything): Demo"
398
- )
399
- description = gr.Markdown(
400
- "## Demo for zero-shot panoptic segmentation using Segment Anything, Grounding DINO, and CLIPSeg."
401
- )
402
- with gr.Row():
403
- with gr.Column():
404
- input_image = gr.Image(source="upload", type="pil")
405
- thing_category_names_string = gr.Textbox(
406
- label="Thing categories (i.e. categories with instances), comma-separated",
407
- placeholder="E.g. car, bus, person",
408
- )
409
- stuff_category_names_string = gr.Textbox(
410
- label="Stuff categories (i.e. categories without instances), comma-separated",
411
- placeholder="E.g. sky, road, buildings",
412
- )
413
- run_button = gr.Button(label="Run")
414
- with gr.Accordion("Advanced options", open=False):
415
- box_threshold = gr.Slider(
416
- label="Grounding DINO box threshold",
417
- minimum=0.0,
418
- maximum=1.0,
419
- value=0.3,
420
- step=0.001,
421
- )
422
- text_threshold = gr.Slider(
423
- label="Grounding DINO text threshold",
424
- minimum=0.0,
425
- maximum=1.0,
426
- value=0.25,
427
- step=0.001,
428
- )
429
- segmentation_background_threshold = gr.Slider(
430
- label="Segmentation background threshold (under this threshold, a pixel is considered background)",
431
- minimum=0.0,
432
- maximum=1.0,
433
- value=0.1,
434
- step=0.001,
435
- )
436
- shrink_kernel_size = gr.Slider(
437
- label="Shrink kernel size (how much to shrink the mask before sampling points)",
438
- minimum=0,
439
- maximum=100,
440
- value=20,
441
- step=1,
442
  )
443
- num_samples_factor = gr.Slider(
444
- label="Number of samples factor (how many points to sample in the largest category)",
445
- minimum=0,
446
- maximum=1000,
447
- value=1000,
448
- step=1,
449
  )
450
-
451
- with gr.Column():
452
- plot = gr.Plot()
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
453
 
454
  examples = gr.Examples(
455
  examples=[
 
393
 
394
  block = gr.Blocks(title="Panoptic Segment Anything").queue()
395
  with block:
396
+ with gr.Column():
397
+ title = gr.Markdown(
398
+ "# [Panoptic Segment Anything](https://github.com/segments-ai/panoptic-segment-anything)"
399
+ )
400
+ description = gr.Markdown(
401
+ "Demo for zero-shot panoptic segmentation using Segment Anything, Grounding DINO, and CLIPSeg."
402
+ )
403
+ with gr.Row():
404
+ with gr.Column():
405
+ input_image = gr.Image(source="upload", type="pil")
406
+ thing_category_names_string = gr.Textbox(
407
+ label="Thing categories (i.e. categories with instances), comma-separated",
408
+ placeholder="E.g. car, bus, person",
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
409
  )
410
+ stuff_category_names_string = gr.Textbox(
411
+ label="Stuff categories (i.e. categories without instances), comma-separated",
412
+ placeholder="E.g. sky, road, buildings",
 
 
 
413
  )
414
+ run_button = gr.Button(label="Run")
415
+ with gr.Accordion("Advanced options", open=False):
416
+ box_threshold = gr.Slider(
417
+ label="Grounding DINO box threshold",
418
+ minimum=0.0,
419
+ maximum=1.0,
420
+ value=0.3,
421
+ step=0.001,
422
+ )
423
+ text_threshold = gr.Slider(
424
+ label="Grounding DINO text threshold",
425
+ minimum=0.0,
426
+ maximum=1.0,
427
+ value=0.25,
428
+ step=0.001,
429
+ )
430
+ segmentation_background_threshold = gr.Slider(
431
+ label="Segmentation background threshold (under this threshold, a pixel is considered background)",
432
+ minimum=0.0,
433
+ maximum=1.0,
434
+ value=0.1,
435
+ step=0.001,
436
+ )
437
+ shrink_kernel_size = gr.Slider(
438
+ label="Shrink kernel size (how much to shrink the mask before sampling points)",
439
+ minimum=0,
440
+ maximum=100,
441
+ value=20,
442
+ step=1,
443
+ )
444
+ num_samples_factor = gr.Slider(
445
+ label="Number of samples factor (how many points to sample in the largest category)",
446
+ minimum=0,
447
+ maximum=1000,
448
+ value=1000,
449
+ step=1,
450
+ )
451
+
452
+ with gr.Column():
453
+ plot = gr.Plot()
454
 
455
  examples = gr.Examples(
456
  examples=[