前言
做电商的朋友都知道,一般平台都会有白底图的要求,在上传白底图后,商品在首页分类页、搜索分类页、品牌馆、电器城等场景中会有更多的露出机会,可能会获得更多的用户流量。
制作白底图本质就是要先抠图,一些抠图质量比较高的网站,比如 https://remove.bg,https://cutout.x-design.com 等,可以做到高质量的抠图。比如,下面的这个图,是属于比较难抠的(为什么难,后面会说):
image.png
在这两个网站的抠图效果如下:
image.png
[图片上传失败…(image-cd46e7-1706760507996)]
image.png
可以看出这张图片的边缘、阴影都被准确识别并抠出了。
那么在 ComfyUI 上是否有对应的算法可以做类似的抠图吗?
首先,很庆幸的是 remove.bg 他开源了 rembg(地址:https://github.com/danielgatis/rembg) 以及一些 u2net 模型,我们可以在 ComfyUI 的管理器中,下载 rembg-comfyui-node 插件,并用这个插件快速的建一个流程,如下:
image.png
我们运行看一下效果:
image.png
看红框部分,惨不忍睹,边缘没有处理干净。这个图的背景加上其他颜色,可以看得更清楚:
image.png
呃,应该是 rembg-comfyui-node 插件作者写得不够好吧? 既然 rembg 已经开源了,我们直接调用 rembg 的原生 API,并尝试调整一些参数看看。 所以,我们封装了一个基于 rembg 的高级版,双击面板,可以添加“背景移除-通用”节点:
image.png
该节点有添加后的效果如下:
image.png
我们可以看到在背景移除中,有 model (模型)可以选择,一般就选择 u2net、除了 u2net 以外,还有其他比如衣服分割的模型。 另外,还有一些高级参数可以设置,比如设置前景分割阈值、后景分割阈值等。我们就先用默认参数尝试一下,如下:
image.png
看来默认参数的运行效果,和 rembg-comfyui-node 插件效果是一致的,红框部分依旧没有处理干净。 那我们尝试一下高级参数,把 alpha_matting 设置为 True,把 foreground_threshold 设置为 200 再看看效果,如下:
image.png
是有一些变化,但效果依然不行。 看来,要想把边缘处理干净,还需要去不断尝试 erode_size、foreground 阈值、background 阈值。这个显然非常费事,而且不同的图片,这几个参数都需要设置成不同的值。
那我们不用 u2net 模型,尝试换成 u2net_human_seg 模型看看? 下面是把模型设置成 u2net_human_seg 后的效果:
image.png
效果依然不好,但比之前默认的 U2NET 模型好一点的是,u2net_human_seg 模型把人的阴影作为一个整体也抠下来了。
显然,用开源的 rembg 来进行图像分割并制作(尤其是批量)白底图,质量完全无法保证。 那为什么 remove.bg 官网他自己抠图效果又那么好呢? 其实道理很简单,他们自己做了模型训练,他们使用的 u2net 模型,不是大家在 github 上免费下载的那个模型。
原力在线经过一番尝试,对于白底图也可以完美的制作了。在原力道场的 ComfyUI 容器中,我们给大家开放了 “背景移除 - 专业版”,具体位置在:
image.png
废话不多说,我们上流程,看效果:
image.png
为了帮助大家更好地掌握 ComfyUI,我在去年花了几个月的时间,撰写并录制了一套ComfyUI的基础教程,共六篇。这套教程详细介绍了选择ComfyUI的理由、其优缺点、下载安装方法、模型与插件的安装、工作流节点和底层逻辑详解、遮罩修改重绘/Inpenting模块以及SDXL工作流手把手搭建。
由于篇幅原因,本文精选几个章节,详细版点击下方卡片免费领取
一、ComfyUI配置指南
- 报错指南
- 环境配置
- 脚本更新
- 后记
- …
二、ComfyUI基础入门
- 软件安装篇
- 插件安装篇
- …
三、 ComfyUI工作流节点/底层逻辑详解
- ComfyUI 基础概念理解
- Stable diffusion 工作原理
- 工作流底层逻辑
- 必备插件补全
- …
四、ComfyUI节点技巧进阶/多模型串联
- 节点进阶详解
- 提词技巧精通
- 多模型节点串联
- …
五、ComfyUI遮罩修改重绘/Inpenting模块详解
- 图像分辨率
- 姿势
- …
六、ComfyUI超实用SDXL工作流手把手搭建
- Refined模型
- SDXL风格化提示词
- SDXL工作流搭建
- …
由于篇幅原因,本文精选几个章节,详细版点击下方卡片免费领取