4.4k stars的抠图高手

2023-03-10 14:44:21 浏览数 (1)

你是怎么扣一个图片的?还在用PS?是时候用一键抠图了。

Rembg 就是你要找的一键抠图工具。

rembg是一款不错的背景移除工具,其采用UNet网络进行训练得到分割模型,精度相当高,本人安装过程遇到一些问题,先简单总结如下:

GitHub数据

  • 4.4k stars
  • 68 watching
  • 719 forks

开源地址:https://github.com/danielgatis/rembg

安装

运行环境Python3.6 以上版本。

CPU版
代码语言:javascript复制
pip install rembg
GPU版本
代码语言:javascript复制
pip install rembg[gpu]

举个例子

远程图像中删除背景
代码语言:javascript复制
> curl -s http://input.png | rembg i > output.png
本地文件中删除背景
代码语言:javascript复制
rembg i path/to/input.png path/to/output.png
批量抠图
代码语言:javascript复制
rembg p path/to/input path/to/output

启动一个抠图服务

我们可以启动一个抠图的服务。作为一个通用的api服务

启动服务器
代码语言:javascript复制
rembg s

也可以用docker运行

代码语言:javascript复制
docker run -p 5000:5000 danielgatis/rembg s

浏览器访问链接直接抠图

代码语言:javascript复制
http://localhost:5000/?url=https://upload.wikimedia.org/wikipedia/commons/thumb/9/9a/Gull_portrait_ca_usa.jpg/1280px-Gull_portrait_ca_usa.jpg
通过浏览器文件上传抠图
代码语言:javascript复制
<form
    action="http://localhost:5000"
    method="post"
    enctype="multipart/form-data"
>
    <input type="file" name="file" />
    <input type="submit" value="upload" />
</form>

更多功能广大网友可以继续挖掘。

0 人点赞