ComfyUI-RemoveBackgroundSuite
基于 ComfyUI 的抠图套件，支持多种抠图模型和细节处理方式。
版本信息
- 当前版本：v0.2
- 更新日期：2024-06-XX
更新日志
v0.2 (2024-06-XX)
- BiRefNetUltraV3_RBS 节点合并模型加载功能，直接选择模型版本，无需单独加载节点。
- BiRefNetUltraV3_RBS、TransparentBackgroundUltra_RBS 节点的 max_megapixels 参数支持动态调整分辨率，自动缩放为32的倍数，防止patch分割报错。
- 新增 Mask Process Details (RBS) 节点，支持VITMatte/GuidedFilter等多种细节处理方法，参数可调。
- 节点命名与界面优化，部分节点重命名。
- 细节处理参数范围优化。
v1.2 (2024-06-XX)
- 节点界面极简化，移除所有细节处理相关参数和功能。
- TransparentBackgroundUltra_RBS 去除 WIP 标志。
v1.1 (2024-03-21)
- 优化 BiRefNet 模型加载逻辑，支持 dynamic、HR、HR-matting 模型
- 优化 VITMatte 模型加载和推理流程
- 改进文档结构和说明
v1.0 (2024-03-20)
- 初始版本发布
- 支持 BiRefNet 和 TransparentBackground 两种抠图模型
节点说明
1. BiRefNet Ultra V3 (RBS)
- 功能：使用 BiRefNet Ultra V3 进行高质量背景移除，支持 dynamic/HR/HR-matting 等多种模型。
- 参数：
image：输入图片（支持批量）。version：选择模型版本（BiRefNet-General、RMBG-2.0、BiRefNet_dynamic等）。device、max_megapixels：详见节点界面。max_megapixels 会自动动态缩放分辨率为32的倍数。
- 输出：
image：去背景后的 RGBA 图片mask：前景掩码
2. Transparent Background Ultra (RBS)
- 功能：将图片背景转换为透明。
- 参数：
image：输入图片。model：选择本地模型。device、max_megapixels：详见节点界面。max_megapixels 会自动动态缩放分辨率为32的倍数。
- 输出：
image：透明背景图片mask：前景掩码
3. Mask Process Details (RBS)
- 功能：对输入图像和mask进行细节优化，支持VITMatte、GuidedFilter等多种方法。
- 参数：
image：输入图片。mask：输入掩码。detail_method：细节处理方法（VITMatte、VITMatte(local)、GuidedFilter）。detail_erode、detail_dilate、black_point、white_point、device、max_megapixels：详见节点界面。
- 输出：
image：优化后的图片mask：优化后的掩码
典型用法
- 用
BiRefNet Ultra V3 (RBS)进行背景移除。 - 可选：用
Mask Process Details (RBS)进一步优化边缘细节。 - 可选：用
Transparent Background Ultra (RBS)处理透明背景。
注意事项
- 请将模型文件放在
ComfyUI/models/transparent-background/目录下，或使用新版节点自动下载。 - 推荐使用 CUDA 设备以获得更快推理速度。
- 细节处理方法对边缘质量有显著影响，可根据实际需求调整。
- 插件所有节点均归类于
RemoveBackgroundSuite，便于统一管理。
依赖安装
pip install -r requirements.txt
常见问题
- 模型下载失败：请检查网络连接或手动下载模型放入指定目录。
- 推理慢/显存不足：可适当降低
max_megapixels或切换到 CPU。 - 节点不显示：请确认插件已放入
custom_nodes目录并重启 ComfyUI。
致谢
本插件大量借鉴和参考了 ComfyUI_LayerStyle_Advance 项目的设计与实现，特别感谢原作者 chflame163 的开源贡献！
如有更多问题请参考原项目文档或在 Issues 区反馈。