欢迎访问我的网站,希望内容对您有用,感兴趣的可以加入免费知识星球。

rembg图片去背景

人工智能 迷途小书童 3年前 (2020-12-30) 8665次浏览 0个评论

软硬件环境

  • windows 10 64bit
  • pytorch 1.7
  • rembg

视频看这里

此处是 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)

您必须 登录 才能发表评论!