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

相关文章

最新工具
Ztalk ai
Ztalk ai

一个AI驱动的会议实时语音翻译平台,可以与 Zoom、Googl...

Supercut
Supercut

一款屏幕录制工具,能帮你快速录下屏幕上的内容,录制的视频可以自动...

DreamShootAI
DreamShootAI

一个能帮你用照片生成各种风格图片的AI情侣照片生成工具,比如情侣...

讯飞星辰MaaS
讯飞星辰MaaS

科大讯飞开发的一款一站式AIGC内容运营平台,主要面向开发者,提...

星辰Agent开发平台
星辰Agent开发平台

讯飞星火新一代智能体Agent开发平台,助力开发者快速搭建生产级...

Drimo智能影视创作平台
Drimo智能影视创作平台

一款由追梦极客推出的面向影视、广告从业者以及AI视频创作者设计的...

码上飞
码上飞

一个需求秒变软件的全流程自动化智能开发平台,码上飞是L4级无人软...

Simular AI
Simular AI

一款专为Mac用户设计的本地AI智能助手,它能通过自然语言处理执...

FotoForensics
FotoForensics

一个由Hacker Factor提供的在线图像篡改检测工具,主要...

PixPin
PixPin

一款功能强大、完全免费的截图、贴图工具,具备高效截图、图片标注、...