前言
做电商的朋友都知道,一般平台都会有白底图的要求,在上传白底图后,商品在首页分类页、搜索分类页、品牌馆、电器城等场景中会有更多的露出机会,可能会获得更多的用户流量。
制作白底图本质就是要先抠图,一些抠图质量比较高的网站,比如 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工作流搭建 …由于篇幅原因,本文精选几个章节,详细版点击下方卡片免费领取
总结
### 文章总结本篇文章主要探讨了电商中白底图的重要性及其制作过程中的抠图挑战,同时介绍了如何利用ComfyUI平台和开源工具`rembg`来实现高质量的自动抠图功能。以下是文章的主要内容总结:
1. **白底图的重要性**:
- 电商平台要求白底图以提升商品的曝光和用户流量。
- 高质量的白底图有助于商品在各类场景中更好地展示。
2. **抠图工具与方法**:
- 介绍了一些高质量的在线抠图网站,如`https://remove.bg`和`https://cutout.x-design.com`,这些工具能识别并抠出复杂图片的边缘及阴影。
- 探讨了ComfyUI平台内使用`rembg-comfyui-node`插件进行抠图的效果,但发现边缘处理不够理想。
3. **rembg的尝试与调整**:
- 详细描述了通过调用rembg的原生API,并尝试调整如`alpha_matting`、`foreground_threshold`、`background_threshold`等高级参数来改善抠图效果。
- 介绍了不同的抠图模型,如`u2net`和`u2net_human_seg`,并对其效果和适用性进行了对比分析。
- 指出开源的rembg模型效果有限,实际使用中往往无法达到商业级别的质量要求。
4. **ComfyUI专业版背景移除功能**:
- 介绍了原力在线在ComfyUI中提供的“背景移除-专业版”功能,该功能通过定制的模型训练,实现了高质量的背景移除效果。
5. **ComfyUI教程推广**:
- 作者花费数月时间撰写并录制了一套ComfyUI的基础教程,共六篇,内容涵盖ComfyUI的配置、基础入门、工作流节点及底层逻辑详解、节点技巧进阶等多个方面。
- 因篇幅限制,文中仅精选了部分章节,完整版教程提供免费领取链接。
### 结论
文章通过实例展示了在ComfyUI平台上使用开源抠图工具进行白底图制作的尝试与挑战,强调了专业版工具对于实现高质量抠图的必要性。同时,通过教程的推广,帮助用户更好地掌握和使用ComfyUI平台。