ComfyUI Extension: ComfyUI-RemoveBackgroundSuite

Authored by whmc76

Created

Updated

0 stars

A matting toolkit based on ComfyUI, supporting multiple matting models and detail processing methods.

Custom Nodes (0)

    README

    ComfyUI-RemoveBackgroundSuite

    基于 ComfyUI 的抠图套件,支持多种抠图模型和细节处理方式。

    版本信息

    • 当前版本:v1.1
    • 更新日期:2024-03-21

    更新日志

    v1.1 (2024-03-21)

    • 优化 BiRefNet 模型加载逻辑,支持 dynamic、HR、HR-matting 模型
    • 改进 BiRefNetUltraV3_RBS 节点,支持多种细节处理方式
    • 优化 VITMatte 模型加载和推理流程
    • 改进文档结构和说明

    v1.0 (2024-03-20)

    • 初始版本发布
    • 支持 BiRefNet 和 TransparentBackground 两种抠图模型
    • 支持多种细节处理方式

    节点说明

    1. LoadBiRefNetModelV3_RBS(推荐)

    • 功能:自动下载并加载 BiRefNet 最新模型,支持 BiRefNet_dynamic。
    • 参数
      • version:选择模型版本(BiRefNet-General、RMBG-2.0、BiRefNet_dynamic)。
    • 输出birefnet_model(供后续节点使用)

    2. BiRefNetUltraV3_RBS(推荐)

    • 功能:使用 BiRefNet Ultra V3 进行高质量背景移除,支持 dynamic 动态模型。
    • 参数
      • image:输入图片(支持批量)。
      • birefnet_model:已加载的模型(支持 dynamic)。
      • detail_methoddetail_erodedetail_dilateblack_pointwhite_pointprocess_detaildevicemax_megapixels:详见节点界面。
    • 输出
      • image:去背景后的 RGBA 图片
      • mask:前景掩码

    3. TransparentBackgroundUltra_RBS([WIP] 开发中)

    ⚠️ 注意:该节点目前处于开发/调试阶段,输出结果可能异常,暂不建议生产环境使用。

    • 功能:将图片背景转换为透明,支持多种细节处理。
    • 参数
      • image:输入图片。
      • model:选择本地模型。
      • 其余参数同上。
    • 输出
      • image:透明背景图片
      • mask:前景掩码

    典型用法

    1. LoadBiRefNetModelV3_RBS 加载模型(推荐选择 BiRefNet_dynamic)。
    2. BiRefNetUltraV3_RBS 进行背景移除。
    3. 可选:用 TransparentBackgroundUltra_RBS 进一步处理透明背景。

    注意事项

    • 请将模型文件放在 ComfyUI/models/transparent-background/ 目录下,或使用新版节点自动下载。
    • 推荐使用 CUDA 设备以获得更快推理速度。
    • 细节处理方法对边缘质量有显著影响,可根据实际需求调整。
    • 插件所有节点均归类于 RemoveBackgroundSuite,便于统一管理。

    依赖安装

    pip install -r requirements.txt
    

    常见问题

    • 模型下载失败:请检查网络连接或手动下载模型放入指定目录。
    • 推理慢/显存不足:可适当降低 max_megapixels 或切换到 CPU。
    • 节点不显示:请确认插件已放入 custom_nodes 目录并重启 ComfyUI。

    致谢

    本插件大量借鉴和参考了 ComfyUI_LayerStyle_Advance 项目的设计与实现,特别感谢原作者 chflame163 的开源贡献!

    如有更多问题请参考原项目文档或在 Issues 区反馈。