中文文档

Support the use of new and old versions of BiRefNet models

Preview

Install

• Manual

    cd custom_nodes
    git clone https://github.com/lldacing/ComfyUI_BiRefNet_ll.git
    cd ComfyUI_BiRefNet_ll
    pip install -r requirements.txt
    # restart ComfyUI

• Via ComfyUI Manager

Models

The available newest models are:

• General: A pre-trained model for general use cases.
• General-Lite: A light pre-trained model for general use cases.
• Portrait: A pre-trained model for human portraits.
• DIS: A pre-trained model for dichotomous image segmentation (DIS).
• HRSOD: A pre-trained model for high-resolution salient object detection (HRSOD).
• COD: A pre-trained model for concealed object detection (COD).
• DIS-TR_TEs: A pre-trained model with massive dataset.

Model files go here (when use AutoDownloadBiRefNetModel automatically downloaded if the folder is not present during first run): models/BiRefNet.

If necessary, they can be downloaded from:

• General ➔ model.safetensors must be renamed General.safetensors
• General-Lite ➔ model.safetensors must be renamed General-Lite.safetensors
• Portrait ➔ model.safetensors must be renamed Portrait.safetensors
• DIS ➔ model.safetensors must be renamed DIS.safetensors
• HRSOD ➔ model.safetensors must be renamed HRSOD.safetensors
• COD ➔ model.safetensors must be renamed COD.safetensors
• DIS-TR_TEs ➔ model.safetensors must be renamed DIS-TR_TEs.safetensors

Some models on GitHub: BiRefNet Releases

Old models:

• BiRefNet-DIS_ep580.pth
• BiRefNet-ep480.pth

Weight Models (Optional)

• swin_large_patch4_window12_384_22kto1k.pth(not General-Lite model)
• swin_tiny_patch4_window7_224_22kto1k_finetune.pth(just General-Lite model)

Nodes

• AutoDownloadBiRefNetModel
  • Automatically download the model into models/BiRefNet, do not support weight model
• LoadRembgByBiRefNetModel
  • Can select model from models/BiRefNet or the path of birefnet configured in the extra YAML file
  • You can download latest models from BiRefNet Releases or old models BiRefNet-DIS_ep580.pth and BiRefNet-ep480.pth
  • When param use_weight is True, need download weight model swin_large_patch4_window12_384_22kto1k.pth
    model General-Lite must use weight model swin_tiny_patch4_window7_224_22kto1k_finetune.pth
• RembgByBiRefNet

Thanks

BiRefNet

dimitribarbot/sd-webui-birefnet