README.md

metadata

title: Segment Anything
emoji: 🚀
colorFrom: gray
colorTo: pink
sdk: gradio
sdk_version: 3.24.1
app_file: app.py
pinned: false

Segment Anything WebUI

This project is based on **Segment Anything Model ** by Meta. The UI is based on Gradio.

• Try deme on HF: AIBoy1993/segment_anything_webui

Change Logs

• [2023-4-11] Support video segmentation.

Usage

Following usage is running on your computer.

• Install Segment Anything（more details about install Segment Anything）：

pip install git+https://github.com/facebookresearch/segment-anything.git

• git clone this repository：

git clone https://github.com/5663015/segment_anything_webui.git

• Make a new folder named checkpoints under this project，and put the downloaded weights files in checkpoints。You can download the weights using following URLs：

  • vit_h: ViT-H SAM model

  • vit_l: ViT-L SAM model

  • vit_b: ViT-B SAM model

• Run：

python app.py

Note： Default model is vit_b，the demo can run on CPU. Default device is cuda。

TODO

• Add segmentation prompt (point and box)

• Add text prompt