ComfyUI-RMBG: 完美人物抠图,背景移除,精准识别分割
ComfyUI-RMBG是什么?
ComfyUI-RMBG 是一个基于 ai 技术的图片处理工具,主要用于背景移除和物体分割。它集成了RMBG-2.0、InSPYRENET 和 BEN 等多种先进的模型,可以高效、精准地完成抠图任务。
ComfyUI-RMBG核心功能:
背景去除(RMBG Node):
能使用多种模型,像RMBG - 2.0、INSPYRENET、BEN、BEN2等。有多种背景选项可选,并且支持批量处理。
RMBG - 2.0模型按照BiRefNet架构构建,在复杂环境里做背景去除时精度高,边缘细节也能保留。
目标分割(Segment Node):
可以通过文本提示来做目标检测,支持标签风格和自然语言输入,利用SAM(Segment Anything Model)实现高精度分割。
另外,这个项目能手动下载和安装不少模型,例如RMBG - 2.0、INSPYRENET、BEN、BEN2、BiRefNet、SAM和GroundingDINO等。这些模型在图像处理场景里各有长处:
RMBG - 2.0:基于BiRefNet架构开发,用于复杂场景的背景提取。有多种背景选项(如透明、黑色、白色),边缘质量优化后抠图更自然。
InSPYRENET:主要针对视频背景去除,批量处理效率高,能调节阈值改善背景效果。
BEN模型:小巧高效,适合算力低的设备,精度较高且计算资源消耗少 。
ComfyUI-RMBG技术要求
需要安装 ComfyUI 和 Python 3.10+。
依赖的 Python 包包括 torch、torchvision、Pillow、numpy 等。
ComfyUI-RMBG安装方法
1. 通过 ComfyUI-Manager 安装
在 ComfyUI-Manager 中搜索 ComfyUI-RMBG 并安装。安装完成后,运行以下命令安装依赖项:
./ComfyUI/python_embeded/python -m pip install -r requirements.txt
2. 手动克隆仓库
将仓库克隆到 ComfyUI 的 custom_nodes 文件夹:
cd ComfyUI/custom_nodes git clone https://github.com/1038lab/ComfyUI-RMBG
3. 安装依赖项。
4. 手动下载模型
首次使用时,模型会自动下载到 ComfyUI/models/RMBG/ 文件夹。
用户也可以手动下载模型文件,并将其放置在指定目录,例如:
RMBG-2.0 模型放置在 /ComfyUI/models/RMBG/RMBG-2.0。
INSPYRENET 模型放置在 /ComfyUI/models/RMBG/INSPYRENET。
其他模型的下载链接和存放路径可参考项目文档。
ComfyUI-RMBG使用场景
电商产品摄影:一键抠图,将产品从复杂背景中提取出来,更换为统一背景。
视频编辑:支持批量处理视频,快速更换背景。
AI 换装和动画制作:结合其他插件实现背景更换和角色换装。
艺术创作:在低算力设备上使用轻量级模型(如 BEN),专注于创意表达。
GitHub 仓库地址:https://github.com/1038lab/ComfyUI-RMBG