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抠图

相关文章

最新工具
ChinaTextbook
ChinaTextbook

一个专门收集和分享中国大陆从小学到大学的教材资源的开源项目,特别...

暗壳AI
暗壳AI

家居AIGC营销与导购工具,它号称“一人即团队”,能够快速生成渲...

UntitledPen
UntitledPen

一个可以生成真实人声旁白的工具,也就是能够在一个平台上编写、编辑...

PageAI
PageAI

一款AI网站生成工具,就像一个小型创意工作室一样,能在短时间内根...

Seedance AI
Seedance AI

字节跳动推出的一款强大的视频生成工具,借助 Seedance 1...

ChatPs
ChatPs

ChatPs 是一款为 Photoshop 设计的插件,它背后是...

趣丸千音
趣丸千音

一个AI语音创作平台,专注提供一站式智能语音解决方案。集成文本转...

太极神器 TAICHI-flet
太极神器 TAICHI-flet

基于flet的一款多功能娱乐软件,实现了浏览图片、音乐、小说、各...

YIKM
YIKM

一个能在线玩小霸王、红白机、街机、FC 等经典游戏的地方,像《魂...

Bridgely
Bridgely

基于AI的实时通用翻译浏览器扩展,支持 60 多种语言,包括 W...