写点什么

如何制作高质量白底图

作者:原力在线
  • 2024-02-01
    江苏
  • 本文字数:1279 字

    阅读完需:约 4 分钟

做电商的朋友都知道,一般平台都会有白底图的要求,在上传白底图后,商品在首页分类页、搜索分类页、品牌馆、电器城等场景中会有更多的露出机会,可能会获得更多的用户流量。


制作白底图本质就是要先抠图,一些抠图质量比较高的网站,比如 https://remove.bghttps://cutout.x-design.com 等,可以做到高质量的抠图。比如,下面的这个图,是属于比较难抠的(为什么难,后面会说):


在这两个网站的抠图效果如下:




可以看出这张图片的边缘、阴影都被准确识别并抠出了。


那么在 ComfyUI 上是否有对应的算法可以做类似的抠图吗?


首先,很庆幸的是 remove.bg 他开源了 rembg(地址:https://github.com/danielgatis/rembg) 以及一些 u2net 模型,我们可以在 ComfyUI 的管理器中,下载 rembg-comfyui-node 插件,并用这个插件快速的建一个流程,如下:



我们运行看一下效果:



看红框部分,惨不忍睹,边缘没有处理干净。这个图的背景加上其他颜色,可以看得更清楚:



呃,应该是 rembg-comfyui-node 插件作者写得不够好吧? 既然 rembg 已经开源了,我们直接调用 rembg 的原生 API,并尝试调整一些参数看看。 所以,我们封装了一个基于 rembg 的高级版,双击面板,可以添加“背景移除-通用”节点:


该节点有添加后的效果如下:



我们可以看到在背景移除中,有 model (模型)可以选择,一般就选择 u2net、除了 u2net 以外,还有其他比如衣服分割的模型。 另外,还有一些高级参数可以设置,比如设置前景分割阈值、后景分割阈值等。我们就先用默认参数尝试一下,如下:



看来默认参数的运行效果,和 rembg-comfyui-node 插件效果是一致的,红框部分依旧没有处理干净。 那我们尝试一下高级参数,把 alpha_matting 设置为 True,把 foreground_threshold 设置为 200 再看看效果,如下:



是有一些变化,但效果依然不行。 看来,要想把边缘处理干净,还需要去不断尝试 erode_size、foreground 阈值、background 阈值。这个显然非常费事,而且不同的图片,这几个参数都需要设置成不同的值。


那我们不用 u2net 模型,尝试换成 u2net_human_seg 模型看看? 下面是把模型设置成 u2net_human_seg 后的效果:



效果依然不好,但比之前默认的 U2NET 模型好一点的是,u2net_human_seg 模型把人的阴影作为一个整体也抠下来了。


显然,用开源的 rembg 来进行图像分割并制作(尤其是批量)白底图,质量完全无法保证。 那为什么 remove.bg 官网他自己抠图效果又那么好呢? 其实道理很简单,他们自己做了模型训练,他们使用的 u2net 模型,不是大家在 github 上免费下载的那个模型。


原力在线经过一番尝试,对于白底图也可以完美的制作了。在原力道场的 ComfyUI 容器中,我们给大家开放了 “背景移除 - 专业版”,具体位置在:


废话不多说,我们上流程,看效果:


效果是不是很赞? 现在登录原力道场:https://www.hiforce.net, 开通 ComfyUI 容器后,你也可以很容易的利用我们的“背景移除 - 专业版”大规模和批量制作白底图了哟!


原力道场 ComfyUI 镜像开通方法,请参考:0、ComfyUI介绍 & 环境开通


本文实例用到的流程文件,可访问原文地址:https://hiforce.yuque.com/org-wiki-hiforce-kbgemz/fpx22q/yf9kvw7zmu2dw9v8 中下载流程文件。

发布于: 刚刚阅读数: 5
用户头像

原力在线

关注

还未添加个人签名 2022-11-25 加入

我们致力于用厚实的数字化基座FOS,打造多元化的SaaS产品,推动企业数字化、政府数字化、城市数字化的进步与融合发展。为客户提供规划、建设、运营服务。 http://www.hiforce.com

评论

发布
暂无评论
如何制作高质量白底图_AI_原力在线_InfoQ写作社区