BiRefNet 是什么?

BiRefNet 是用于背景去除和抠图的模型。

正式来说,它是用于 Dichotomous Image Segmentation (DIS) 的模型,也就是把图像分成 前景背景 两部分。

它并不是像 SAM 那样,通过“这个点”“这个框”“这个物体”来指定并生成蒙版。BiRefNet 更擅长从头发、植物等细节复杂的对象中,生成高质量的切出蒙版。


模型下载

📂ComfyUI/
└── 📂models/
    └── 📂background_removal/
        └── birefnet.safetensors

workflow

切出前景

BiRefNet.json
  • 使用 Load Background Removal Model 读取 birefnet.safetensors
  • 将图像和模型输入 Remove Background 后,会输出用于背景去除的 MASK
  • 想要背景透明的图像时,使用 Join Image with Alpha
    • 但如果直接使用这个蒙版,前景会变透明,所以中间要接一个 Invert Mask

填充背景

上面的 workflow 会把背景变为透明,但作为图像生成或分析的预处理使用时,把背景填充成单色通常更容易处理。

BiRefNet_fill.json
  • 将原图、反转后的蒙版、单色图像输入 Image Composite Masked
  • 只把被蒙版选中的背景部分替换为单色图像。
  • 详情请参考 图层合成