sd-webui-segment-anything

环境要求

• 已安装 AUTOMATIC1111 Stable Diffusion WebUI（插件宿主）
• 足够的磁盘空间存放 SAM 权重（375MB~2.57GB，按所选模型而定）
• 显存不足时需改用更小的模型（作者在 NVIDIA 3090 Ti 上验证了 vit_h）
• 可访问 dl.fbaipublicfiles.com 或 huggingface.co 下载模型

安装与启动步骤

1. 1进入扩展目录

   定位到 Stable Diffusion WebUI 的 extensions 目录，插件需要放在这里才能被识别。

   cd /your/path/stable-diffusion-webui/extensions

   复制

2. 2克隆插件仓库

   用 git clone 把 sd-webui-segment-anything 拉取到 extensions 目录，也可以在 WebUI 界面里安装。

   git clone https://github.com/continue-revolution/sd-webui-segment-anything.git

   复制

3. 3准备模型目录

   在 sd-webui 的 models 下新建 sam 目录存放权重；若改用扩展自带 models/sam，则需删掉这个目录，二者只能选一个。

   mkdir -p /your/path/stable-diffusion-webui/models/sam

   复制

4. 4下载 SAM 权重

   任选一个模型下载：sam_vit_h(2.56GB)、sam_vit_l(1.25GB)、sam_vit_b(375MB)，显存不够就选更小的。

   cd /your/path/stable-diffusion-webui/models/sam
   wget https://dl.fbaipublicfiles.com/segment_anything/sam_vit_h_4b8939.pth

   复制

5. 5可选换其它模型

   也可改用 SAM-HQ 的 sam_hq_vit_h/l/b，下载后同样放进 sam 目录。

   wget https://huggingface.co/lkeab/hq-sam/resolve/main/sam_hq_vit_b.pth

   复制

6. 6检查模型文件名

   必须保持下载时的原始文件名，改名会导致 segment anything 内部报错、插件失效。

7. 7重启 WebUI

   重启 Stable Diffusion WebUI，扩展会加载 SAM 模型；首次使用 GroundingDINO 时会自动安装依赖和模型。

关键配置

如何确认成功

重启 WebUI 后能看到 Segment Anything 扩展界面，上传图片点击画面即可生成分割掩码。

常见问题

注意事项

• 模型目录只能选一处：用扩展内的 models/sam 时，需删除 ${sd-webui}/models/sam。
• GroundingDINO 的包、模型以及 ControlNet annotator 模型会在首次使用时自动安装。
• Matting-Anything 与 FastSAM 目前尚未支持，README 中标注为计划支持。
• 首次下载权重体积较大，请预留磁盘空间并保持网络稳定。