BiRefNet

节点功能:这个节点加载 Hugging Face 上的 RMBG-2.0 模型(默认为 briaai/RMBG-2.0,但是该模型需要统一协议不然没办法下载),并输出一个“去背景函数” remove_bg_fn(image),可用于图像的前景提取。(建议加载ZhengPeng7/BiRefNet模型)模型会自动下载在C:\Users\(自己用户名)\.cache\huggingface\hub目录下面,要保证网络没问题。

输出参数

参数名称说明
FUNCTION背景移除函数(可传递给其他节点如 ImagePreprocessor 的 remove_bg_fn 输入)。

控件参数

参数名称说明
ckpt_nameHuggingFace 上背景分割模型的仓库名或路径,默认值为 "briaai/RMBG-2.0"。

如果加载briaai/RMBG-2.0模型会报错如下信息。

下图为BiRefNet模型加载并且抠图效果演示。