README.md

Segment Anything WebUI

[Paper] [Project] [Demo] [Dataset] [Blog]

本项目是 Meta 最新成果 Segment Anything Model 的简单 WebUI，基于 Gradio。虽然官方有很好的 demo，但也算是通过这么个简单的 UI 熟悉一下这个成果的用法。目前只实现了自动分割。

用法：

• 安装 Segment Anything（其他安装方式）：

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

• git clone 本仓库：

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

• 在项目目录下新建 checkpoints 文件夹，将下载的模型文件放进去。模型文件下载：

  • default or vit_h: ViT-H SAM model.

  • vit_l: ViT-L SAM model.

  • vit_b: ViT-B SAM model.

• 运行：

python app.py

注意： 默认模型是 vit_b，笔记本 CPU 可运行。运行硬件默认是 cuda。

TODO

• 增加 segmentation prompt