ComfyUI Segment Anything

本项目是 https://github.com/continue-revolution/sd-webui-segment-anything 的 ComfyUI 版本。目前仅实现了最核心的功能。感谢 continue-revolution 所做的先驱工作，本项目正是基于他们的成果开发而成。

在输入相同的情况下，我已确保本项目的输出与 sd-webui-segment-anything 保持一致。

环境要求

请确保通过以下命令安装了 Python 依赖：

pip3 install -r requirements.txt

模型

模型在使用时会自动下载。您也可以根据下表手动下载。如果自动下载速度较慢，您可以设置 HTTP_PROXY 和 HTTPS_PROXY 环境变量来使用代理。

bert-base-uncased

您可以从 https://huggingface.co/bert-base-uncased/tree/main 下载该模型，并将其放入 ComfyUI 根目录下的 models/bert-base-uncased 文件夹中，具体如下：

ComfyUI
    models
        bert-base-uncased
            config.json
            model.safetensors
            tokenizer_config.json
            tokenizer.json
            vocab.txt

您也可以跳过此步骤。在推理过程中，bert-base-uncased 将通过 transformers 库自动下载，其目录通常为 ~/.cache/huggingface/hub/models--bert-base-uncased。

GroundingDino

请直接将模型和配置文件下载到 ComfyUI 根目录下的 models/grounding-dino 目录，不要修改文件名。

名称	大小	配置文件	模型文件
GroundingDINO_SwinT_OGC	694MB	下载链接	下载链接
GroundingDINO_SwinB	938MB	下载链接	下载链接

SAM

请直接将模型文件下载到 ComfyUI 根目录下的 models/sams 目录，不要修改文件名。

名称	大小	模型文件
sam_vit_h	2.56GB	下载链接
sam_vit_l	1.25GB	下载链接
sam_vit_b	375MB	下载链接
sam_hq_vit_h	2.57GB	下载链接
sam_hq_vit_l	1.25GB	下载链接
sam_hq_vit_b	379MB	下载链接
mobile_sam	39MB	下载链接

贡献

感谢您考虑为源代码提供帮助！欢迎互联网上的任何人贡献，即使是最小的修复，我们也深表感谢！

如果您想为此项目做出贡献，请进行分支（fork）、修复、提交（commit）并发送拉取请求（pull request），以便我审核并合并到主代码库中。