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)实现高精度分割。

ComfyUI-RMBG核心功能.webp

另外,这个项目能手动下载和安装不少模型,例如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抠图.webp

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),专注于创意表达。

ComfyUI-RMBG使用场景.webp

GitHub 仓库地址:https://github.com/1038lab/ComfyUI-RMBG

收藏
#抠图软件 #ai抠图

相关文章

最新工具
GenAI_Agents
GenAI_Agents

一个专注于生成式AI Agents技术的开源项目,GenAI_A...

GPTBots AI
GPTBots AI

GPTBots是一个无需编写代码就能构建AI Agent的平台。...

SwiftChat
SwiftChat

一款基于React Native开发的快速、安全、跨平台聊天应用...

Duck AI
Duck AI

DuckDuckGo推出的免费、匿名AI聊天机器人,用户可以放心...

Glean AI Agent
Glean AI Agent

Glean推出的企业级实用平台,能借助自然语言创建并部署AI代理...

Awesome LLM Apps
Awesome LLM Apps

一个精选的LLM(大型语言模型)应用集合,汇集了使用OpenAI...

Browser-use
Browser-use

一个开源的Python库,它把AI技术和浏览器自动化功能结合起来...

NotaGen
NotaGen

由中央音乐学院、北京航空航天大学和清华大学等机构联合开发的一款 ...

Stagetimer.io
Stagetimer.io

一款由Lukas Hermann开发的远程控制倒计时计时器工具,...

MetaGPT
MetaGPT

一个开源的多智能体框架,由DeepWisdom团队开发,在202...