sam-at-home

A gradio interface to explore capabilities of Meta AI's Segment Anything Model (SAM).

Me: Mom, can we have SAM?
Mom: No there is SAM at home.
SAM at home:

Automatic Mask Generator Interface

Tool to interact with the SamAutomaticMaskGenerator interface. Specify parameters to control the automatic generation. Optionally extract transparent images of each detected segment for saving.

Segmentation of Sam the cat using Sam the model

Predict Interface (experimental)

There is also minor support for foreground and background keypoints on the image to guide the prediction. This is done on a best effort basis by using the sketch-color canvas of gradio, trying to translate brush strokes into keypoints and bounding boxes. For the latter this is only supported for one bounding box currently. Has a few performance issues right now with the canvas being slow/fixed to a small size, and anti-aliasing of brush strokes causing edge cases in the box detection algorithm. This should hopefully improve in the future as gradio adds UI components for bounding boxes/keypoints.

Guided Prediction Interface

Setup

1. Set up virtual environment using the script ./firstTimeSetup.sh, or with whichever virtual environment tool you prefer.
2. Activate virtual environment source venv/bin/activate
3. Download models (see below)
4. Run UI: gradio app.py or make run
5. Load UI in browser, usually on localhost:7860

Downloading Models

Models can be downloaded using the download_models.py script. By default gets the vit_h model, with the option to download all to the models folder. Alternatively you can fetch them directly from the official repository and manually place them there.

usage: Model downloader for the Segment Anything Models [-h] [--model {all,vit_h,vit_b,vit_l}]

optional arguments:
  -h, --help            show this help message and exit
  --model {all,vit_h,vit_b,vit_l}

Citations

@article{kirillov2023segany,
  title={Segment Anything},
  author={Kirillov, Alexander and Mintun, Eric and Ravi, Nikhila and Mao, Hanzi and Rolland, Chloe and Gustafson, Laura and Xiao, Tete and Whitehead, Spencer and Berg, Alexander C. and Lo, Wan-Yen and Doll{\'a}r, Piotr and Girshick, Ross},
  journal={arXiv:2304.02643},
  year={2023}