ComfyUI-BiRefNet-Super

English | 中文

Introduction

This repository packages the latest BiRefNet model as a ComfyUI node for use, supporting chunked loading on both CPU and GPU, as well as model caching features.

Features

Feature 1: Supports chunked loading on CPU and GPU

When CUDA is enabled, specify the cpu_size to load part of the model onto the CPU.


Feature 2: Model Caching

Feature 3: Multiple Cropping Methods

Supports putalpha, naive, and alpha_matting cropping methods.

News

• Nov 19, 2024: Add mask_precision_threshold parameter to control the accuracy threshold of the mask, default to 0.1
  
• Dec 10, 2024:
  • Add BiRefNet_onnx node to Support onnx model
    
  • Modify the name of repository from ComfyUI-BiRefNet-lite to ComfyUI-BiRefNet-Super
    

Installation

Method 1:

1. Navigate to the node directory, ComfyUI/custom_nodes/
2. git clone https://github.com/rubi-du/ComfyUI-BiRefNet-Super.git
3. cd ComfyUI-BiRefNet-Super
4. pip install -r requirements.txt
5. Restart ComfyUI

Method 2:

Directly download the node source code package, unzip it into the custom_nodes directory, and then restart ComfyUI.

Method 3:

Install via ComfyUI-Manager by searching for "ComfyUI-BiRefNet-Super".

Usage

BiRefNet_Super | BiRefNet_Lite

Example workflows are placed in ComfyUI-BiRefNet-Super/workflow.

There are two options for loading models: one is to automatically download and load a remote model, and the other is to load a local model (in which case you need to set load_local_model to true and set local_model_path to the local model path under the models/birefnet directory, for example, the BiRefNet folder).

---

Model download links:

BiRefNet: https://huggingface.co/ZhengPeng7/BiRefNet/tree/main

BiRefNet_lite-2K: https://huggingface.co/ZhengPeng7/BiRefNet_lite-2K/tree/main

BiRefNet-portrait: https://huggingface.co/ZhengPeng7/BiRefNet-portrait/tree/main

BiRefNet-matting: https://huggingface.co/ZhengPeng7/BiRefNet-matting/tree/main

RMBG-2.0: https://huggingface.co/briaai/RMBG-2.0/tree/main

---

Usage of workflow.json

---

Usage of video_workflow.json

Workflow Address

BiRefNet_onnx

Example workflows are placed in workflow.

Place the model file in ComfyUI/models/birefnet.

Model file name should be one of the following:

• birefnet-general.onnx
• birefnet-general-lite.onnx
• birefnet-portrait.onnx
• birefnet_massive.onnx
• birefnet-hrsod.onnx
• birefnet-dis.onnx
• birefnet-cod.onnx

Model download links:

birefnet-genernal: https://github.com/danielgatis/rembg/releases/download/v0.0.0/BiRefNet-general-epoch_244.onnx
birefnet-genernal-lite: https://github.com/danielgatis/rembg/releases/download/v0.0.0/BiRefNet-general-bb_swin_v1_tiny-epoch_232.onnx
birefnet-portrait: https://github.com/danielgatis/rembg/releases/download/v0.0.0/BiRefNet-portrait-epoch_150.onnx
birefnet_massive: https://github.com/danielgatis/rembg/releases/download/v0.0.0/BiRefNet-massive-TR_DIS5K_TR_TEs-epoch_420.onnx
birefnet-hrsod: https://github.com/danielgatis/rembg/releases/download/v0.0.0/BiRefNet-HRSOD_DHU-epoch_115.onnx
birefnet-dis: https://github.com/danielgatis/rembg/releases/download/v0.0.0/BiRefNet-DIS-epoch_590.onnx
birefnet-cod: https://github.com/danielgatis/rembg/releases/download/v0.0.0/BiRefNet-COD-epoch_125.onnx

Acknowledgments

Thanks to all the authors of the BiRefNet repository ZhengPeng7/BiRefNet.

Some code was referenced from MoonHugo/ComfyUI-BiRefNet-Hugo. Thanks!

Some code was referenced from danielgatis/rembg. Thanks!