参数名称 | 说明 |
---|---|
FUNCTION | 背景移除函数(可传递给其他节点如 ImagePreprocessor 的 remove_bg_fn 输入)。 |
BiRefNet
节点功能:这个节点加载 Hugging Face 上的 RMBG-2.0 模型(默认为 briaai/RMBG-2.0,但是该模型需要统一协议不然没办法下载),并输出一个“去背景函数” remove_bg_fn(image),可用于图像的前景提取。(建议加载ZhengPeng7/BiRefNet模型)模型会自动下载在C:\Users\(自己用户名)\.cache\huggingface\hub目录下面,要保证网络没问题。
输出参数
控件参数
参数名称 | 说明 |
---|---|
ckpt_name | HuggingFace 上背景分割模型的仓库名或路径,默认值为 "briaai/RMBG-2.0"。 |
如果加载briaai/RMBG-2.0模型会报错如下信息。
下图为BiRefNet模型加载并且抠图效果演示。