AI图片去背景
软硬件环境 简介
rembg是一个去除图片背景的工具,不仅可以在中调用,而且也提供了可执行程序,非常的方便。
安装
直接通过pip安装
pip install rembg
或者从拉取源码进行安装
git clone https://github.com/danielgatis/rembg.git
cd rembg
pip install -r requirements.txt
安装完成后,会生成rembg可执行文件,后面就可以用它来测试了
使用命令行
首先下载模型文件,这里提供2个地址:
文件下载下来后,拷贝到~/.u2net文件夹下,默认这个文件夹不存在,需要自行创建
接下来就可以使用本地图片进行测试了
rembg -o output_test.jpg test.jpg
-o是处理后的图片
rembg
rembg
rembg
rembg
除了处理单张图片外,rembg还支持对文件夹的处理,这里将上面的2张图片放在一个文件夹中,然后使用命令
rembg -p test
rembg
rembg还提供了web端的接口,执行rembg-启动后台服务,这里面使用的是flask框架
rembg
然后在浏览器中输入
http://127.0.0.1:5000/?url=https://image.xugaoxiang.com/imgs/2020/12/bf5f5553a596319b.jpg
url=后面的是图片的地址,可以看出rembg是支持https的
rembg 代码中调用rembg
看下面示例
# app.py
import sys
from rembg.bg import removesys.stdout.buffer.write(remove(sys.stdin.buffer.read()))
然后在终端中执行
cat bryant.jpg | python app.py > output.png
代码运行的结果跟上面是一样的
参考资料