加微信进交流群:xituxiaoshutong100

rembg图片去背景

ML/DL 迷途小书童 0评论

软硬件环境

视频看这里

此处是youtube的播放链接,需要科学上网。喜欢我的视频,请记得订阅我的频道,打开旁边的小铃铛,点赞并分享,感谢您的支持。

简介

rembg是一个去除图片背景的python工具,不仅可以在python中调用,而且也提供了可执行程序,非常的方便。

安装

直接通过pip安装

pip install rembg

或者从github拉取源码进行安装

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-server启动后台服务,这里面使用的是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 remove

sys.stdout.buffer.write(remove(sys.stdin.buffer.read()))

然后在终端中执行

cat bryant.jpg | python app.py > output.png

代码运行的结果跟上面是一样的

参考资料

喜欢 (1)
发表我的评论
取消评论

表情