摘要:因为我最近需要用到 AI抠图,所以花了 2 天时间把目前比较主流,又可以通过 docker 快速部署的 AI 抠图工具都测试了一下。这里就来聊聊我测试的这些容器哪个最好用。
因为我最近需要用到 AI抠图,所以花了 2 天时间把目前比较主流,又可以通过 docker 快速部署的 AI 抠图工具都测试了一下。这里就来聊聊我测试的这些容器哪个最好用。
其实 AI抠图效果主要取决于 AI大模型,目前比较知名的抠图模型有:BiRefNet2.0、RMBG2.0、BEN 等。如果你想集成部署多个 AI 抠图工具,我推荐选择 ComfyUI 来操作。但由于我需要的是短平快的工具,所以选择了容器化部署。
首先介绍是的 ihmily/image-matting
image-matting 基于 briaai/RMBG-1.4 模型(建议作者能升级成 RMBG-2.0)。首先通过 docker-compose 来部署 ihmily/image-matting。
services: image-matting: ports: - 7071:8000 image: ihmily/image-matting:latest部署完毕后访问 image-matting 主页,功能非常精简:上传图片,然后选择通用抠图模式或人像抠图模式,上传完毕自动抠图。image-matting 一大优点是:不仅会生成抠图图片,还会生成蒙版图片。
下面我用两张图片测试了一下效果:一张物品、一张人像。image-matting 的抠图效果和速度都听不错的。
接下来介绍的是 danielgatis/rembg
rembg 使用的是 rembg 算法,我们同样使用 docker-compose 来部署danielgatis/rembg。
services: rembg: ports: - 7070:7000 restart: always image: danielgatis/rembg command: srembg 的主页选项中可以手动调节抠图参数,但是 rembg 实际使用中却并不好用,容易出现抠图不完整或者抠图太过了的情况,而且速度也不算快。我个人不太推荐这个容器。
接下来是 showtim2007/modelscope
modelscope 是阿里巴巴达摩院推出的一个AI工具集,我们用到的是抠图功能。同样通过 docker-compose 部署 showtim2007/modelscope。
services: modelscope: container_name: BSHM ports: - 5050:5000 image: showtim2007/modelscope:1.4.3 command: bash -c "cd /opt &&python img.py"modelscope 的界面同样很精简,首先上传图像然后选择人像抠图或者通用抠图即可。抠图效果和 ihmily/image-matting 差不多。
人像抠图 HivisionIDPhotos
linzeyi/hivision_idphotos 是一个人像抠图工具,我之前已经介绍过。hivision_idphotos 可以选择多个抠图算法。比如:modnet、RMBG-1.4、BiRefNet1.0 等,尤其是还可以一键输出证件照尺寸。可以说是目前最好用的人像抠图容器。
version: '3.8'services: hivision_idphotos: ports: - 7860:7860 image: linzeyi/hivision_idphotos自己部署 RMBG2.0
RMBG2.0 算法 11月才发布,目前还没找到整合好现成能用的容器。所以我基于 Gradio 自己 build 了一个容器。
RMBG2.0 抠图效果确实听不错的,我目前用的就是这个自己部署的这个容器在使用。
不得不说随着算法模型迭代,效果确实越来越好了,相信过一段时间这些容器都会升级算法。目前的话人像抠图我会推荐 HivisionIDPhotos ,普通抠图我会推荐 image-matting,或者你可以像我一样自己部署 RMBG2.0。
现在用AI抠图效果比人手动抠图还要细致,确实让我省了不少时间。
来源:肖宇科技频道