简介
RMBG
旨在有效地将各种背景中的前景与背景分离 类别和图像类型。
该模型已经在精心挑选的数据集上进行了训练,其中包括: 一般库存图像、电子商务、游戏和广告内容,使其适合支持大规模企业内容创建的商业用例。
其准确性、效率和多功能性可与目前领先的可用源模型相媲美。 当内容安全、合法许可的数据集和偏见缓解至关重要时,它是理想的选择。
Bria-RMBG
模型使用超过 15,000 张高质量、高分辨率、手动标记(像素精度)、完全许可的图像进行训练。 我们的基准包括平衡的性别、平衡的种族和不同类型的残疾人。
RMBG-2.0
是在BiRefNet架构上开发的,并通过我们专有的数据集和训练方案进行了增强。该训练数据显着提高了模型背景去除任务的准确性和有效性。
模型对比:
在线Demo:
https://huggingface.co/spaces/briaai/BRIA-RMBG-2.0
ComfyUI 中安装 RMBG2.0
插件安装
目前支持RMBG2.0模型的有三个插件:
- • https://github.com/chflame163/ComfyUI_LayerStyle 集成众多功能,推荐用,但是比较难装
- • https://github.com/1038lab/ComfyUI-RMBG 功能单一,安装简单
- • https://github.com/gorillaframeai/GF_nodes 功能单一,安装简单
对于LayerStyle的安装,请参考官方Github,上面有明确的中文说明:
https://github.com/chflame163/ComfyUI_LayerStyle/blob/main/README_CN.MD#%E5%AE%89%E8%A3%85%E6%96%B9%E6%B3%95
这里不再赘述。
别忘了给几位开源大佬点个Star支持一下。
安装插件:以上都可以通过manager来安装,搜索RMBG
即可,如下图:ZHO 大佬的还是1.4旧版本,下面两个任意选择一个即可。
模型下载
模型都是同一个模型,只是这三个插件中,模型存放的位置不同。
对于chflame163/ComfyUI_LayerStyle 插件
从 huggingface 或 百度网盘(百度网盘) 下载全部文件并复制到 ComfyUI/models/BiRefNet/RMBG-2.0
文件夹。
对于1038lab/ComfyUI-RMBG 插件
插件会自动下载,默认是从Huggingface下载,并且缓存到comfyUI/models/RMBG/RMBG-2.0
目录下,也可以通过链接解决。
对于gorillaframeai/GF_nodes 插件
从 huggingface 或 百度网盘(百度网盘) 下载全部文件并复制到 ComfyUI/models/RMBG-2.0
文件夹
如何解决三个插件共用同一个模型呢:
# 实际模型存放位置:ComfyUI/models/BiRefNet/RMBG-2.0
# 对于gorillaframeai/GF_nodes 插件
cd ComfyUI/models
mklink /d "RMBG-2.0" ".\BiRefNet\RMBG-2.0"
# 对于1038lab/ComfyUI-RMBG 插件
cd D:\SD\ComfyUI-aki-v1.4\models\rmbg\RMBG-2.0\models--briaai--RMBG-2.0\snapshots\79cdc4751cdf5b9570eb7c6a8bfa7fb04bf10b6c
rm -rf model.safetensors
mklink model.safetensors D:\SD\ComfyUI-aki-v1.4\models\RMBG-2.0\model.safetensors
目录结构如下:
原仓库中的,只需下载如下模型,可以节省很多磁盘空间:
ComfyUI RMBG2.0 工作流
原图 | 抠图 | 蒙版 |
| | |
| | |