https://github.com/rubi-du/ComfyUI-BiRefNet-Super

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.
https://github.com/rubi-du/ComfyUI-BiRefNet-Super

birefnet comfyui onnx remove-background removebg rmbg segment segmentation

Last synced: about 2 months ago
JSON representation

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.

• Host: GitHub
• URL: https://github.com/rubi-du/ComfyUI-BiRefNet-Super
• Owner: rubi-du
• License: gpl-3.0
• Created: 2024-11-18T04:57:50.000Z (11 months ago)
• Default Branch: main
• Last Pushed: 2025-05-21T02:21:09.000Z (5 months ago)
• Last Synced: 2025-05-21T03:29:08.041Z (5 months ago)
• Topics: birefnet, comfyui, onnx, remove-background, removebg, rmbg, segment, segmentation
• Language: Python
• Homepage:
• Size: 7.8 MB
• Stars: 9
• Watchers: 1
• Forks: 4
• Open Issues: 1
• Metadata Files:
  • Readme: README.md
  • License: LICENSE

Awesome Lists containing this project

• awesome-comfyui - **ComfyUI-BiRefNet-lite**
• awesome-comfyui - **ComfyUI-BiRefNet-lite**
• awesome-comfyui - **ComfyUI-BiRefNet-lite**

README

          
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.  

![slot](./assets/feature1.png)
  

Feature 2: Model Caching
  

Feature 3: Multiple Cropping Methods
  

Supports putalpha, naive, and alpha_matting cropping methods.  

![slot](./assets/feature2.png)
  

  

## 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).
  

  

![](./assets/9e6bf0f9-67a7-41ea-bc4b-d8352e4fac4a.png)  

___  

  

![](./assets/model_path.png)  

  

#### 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
  

  

![plot](./assets/demo1.png)  

  

___  

Usage of video_workflow.json
  

[Workflow Address](./workflow/video_workflow.json) 

### BiRefNet_onnx

Example workflows are placed in [workflow](./workflow/workflow-onnx.json).


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

![plot](./assets/demo2.png)  

#### 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](https://github.com/zhengpeng7/birefnet).  

  

Some code was referenced from [MoonHugo/ComfyUI-BiRefNet-Hugo](https://github.com/MoonHugo/ComfyUI-BiRefNet-Hugo). Thanks!

Some code was referenced from [danielgatis/rembg](https://github.com/danielgatis/rembg). Thanks!