ComfyUI Segment Anything图像分割工具完整安装指南
项目地址: https://gitcode.com/gh_mirrors/co/comfyui_segment_anything ComfyUI Segment Anything是一款基于GroundingDino和SAM技术的开源图像分割工具，能够通过简单的语义描述精确分割图像中的任意元素。作为sd-webui-segment-anything项目的ComfyUI版本，目前已经实现了最核心的图像语义分割功能。
项目环境要求
在开始安装之前，请确保您的系统满足以下基本要求：
- Python 3.8或更高版本
- pip包管理工具
- 足够的磁盘空间用于存储模型文件
完整安装步骤
获取项目源码
首先需要将项目克隆到本地：
git clone https://gitcode.com/gh_mirrors/co/comfyui_segment_anything
安装Python依赖
进入项目目录后，执行以下命令安装所需的Python依赖包：
pip3 install -r requirements.txt
模型文件配置
项目会自动下载所需的AI模型文件。如果自动下载速度较慢，您也可以手动配置模型文件：
BERT基础模型配置 将bert-base-uncased模型文件放置在ComfyUI根目录的models/bert-base-uncased文件夹中，包含以下文件：
- config.json
- model.safetensors
- tokenizer_config.json
- tokenizer.json
- vocab.txt
GroundingDino模型选项 项目支持两种GroundingDino模型：
- GroundingDINO_SwinT_OGC（694MB）
- GroundingDINO_SwinB（938MB）
SAM模型系列 提供多种SAM模型供选择：
- 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）
网络优化设置
如果自动下载模型速度较慢，可以通过设置代理环境变量来加速下载：
export HTTP_PROXY=http://your-proxy-server:port
export HTTPS_PROXY=http://your-proxy-server:port
核心功能使用
项目的主要功能集中在node.py脚本中，通过以下命令即可启动图像分割功能：
python3 node.py
该脚本将根据输入的图像和语义描述，自动分割图像中的指定元素并输出结果。项目在输入相同的情况下，输出与sd-webui-segment-anything保持一致。
项目结构说明
项目的核心代码组织在comfyui_segment_anything目录下，包含以下主要模块：
- local_groundingdino：GroundingDino检测器的本地实现
- sam_hq：高质量SAM分割模型
- node.py：主要处理脚本
使用技巧与最佳实践
为了获得最佳的图像分割效果，建议使用清晰、具体的语义描述。例如，"分割图像中的人"、"提取红色的汽车"等明确的指令能够帮助模型更准确地理解您的需求。
通过以上完整的安装和使用指南，您现在可以开始体验ComfyUI Segment Anything强大的图像分割功能，为您的图像处理工作流程增添新的AI能力。
创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考
