因为我最近需要用到 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: s
rembg 的主页选项中可以手动调节抠图参数,但是 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 了一个容器。
https://github.com/94486/RMBG2.0-docker
模型文件可以在这个地址进行下载 https://www.modelscope.cn/models/AI-ModelScope/RMBG-2.0/files
RMBG2.0 抠图效果确实听不错的,我目前用的就是这个自己部署的这个容器在使用。
不得不说随着算法模型迭代,效果确实越来越好了,相信过一段时间这些容器都会升级算法。目前的话人像抠图我会推荐 HivisionIDPhotos ,普通抠图我会推荐 image-matting,或者你可以像我一样自己部署 RMBG2.0。
现在用AI抠图效果比人手动抠图还要细致,确实让我省了不少时间。