AI抠图容器究竟哪个最好用

B站影视 2024-12-26 06:13 1

摘要:因为我最近需要用到 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: 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 了一个容器。

RMBG2.0 抠图效果确实听不错的,我目前用的就是这个自己部署的这个容器在使用。

不得不说随着算法模型迭代,效果确实越来越好了,相信过一段时间这些容器都会升级算法。目前的话人像抠图我会推荐 HivisionIDPhotos ,普通抠图我会推荐 image-matting,或者你可以像我一样自己部署 RMBG2.0。

现在用AI抠图效果比人手动抠图还要细致,确实让我省了不少时间。

来源:肖宇科技频道

相关推荐