comfyui_segment_anything

ComfyUI Segment Anything 项目介绍

ComfyUI Segment Anything 项目是一个基于 ComfyUI 的图像分割工具，受到 sd-webui-segment-anything 项目的启发并与之保持一致。尽管目前仅实现了该工具的核心功能，但开发者仍致力于在同样的输入条件下，保持与原有项目一致的输出效果。

项目依赖要求

要顺利运行 ComfyUI Segment Anything，首先需要确保系统中已安装所需的 Python 依赖。可以通过以下命令来安装：

pip3 install -r requirements.txt

模型文件下载

项目使用了一些预训练模型文件以实现图像分割功能。这些模型会在首次使用时自动下载，但如果自动下载速度较慢，用户可以设置 HTTP_PROXY 和 HTTPS_PROXY 环境变量，通过代理进行加速。此外，用户也可以选择手动下载模型文件。

bert-base-uncased

• 模型文件来自 Hugging Face 的 bert-base-uncased。用户可将文件下载至 ComfyUI 的根目录下的 models/bert-base-uncased 文件夹。若跳过手动下载，transformers 库会在推理过程中自动进行下载，默认目录为 ~/.cache/huggingface/hub/models--bert-base-uncased。

GroundingDino

• 此模型的配置文件和权重文件需直接下载至 ComfyUI 根目录下的 models/grounding-dino 文件夹，并保持文件名不变。
• 可选择的模型：
  • GroundingDINO_SwinT_OGC：文件大小 694MB。
  • GroundingDINO_SwinB：文件大小 938MB。

SAM 模型

• SAM 的模型文件同样需要直接下载至 ComfyUI 根目录下的 models/sams 文件夹，文件名无须修改。
• 可供选择的不同规模的模型，包括 sam_vit_h、sam_vit_l、sam_vit_b 等，其中不同模型的文件大小不一，譬如 sam_vit_h 的大小为 2.56GB。

贡献与合作

项目欢迎任何形式的贡献。无论是代码的小修小改，还是功能的新添，都能使项目更趋完善。若希望参与项目开发，可以先 fork 本项目，完成修改后提交 pull request，开发者将审核并合并至主干代码库。

总之，ComfyUI Segment Anything 项目为用户提供了一种强大且有效的图像分割工具，并不断向社区开放以接受更多的改进和优化。