软硬件环境
- 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个地址：
-
百度网盘链接：https://pan.baidu.com/s/1IJOZiELPL9KJ8andi9fpyA，提取码：
lgug
文件下载下来后，拷贝到 ~/.u2net 文件夹下，默认这个文件夹不存在，需要自行创建
接下来就可以使用本地图片进行测试了
rembg -o output_test.jpg test.jpg-o 是处理后的图片
除了处理单张图片外，rembg 还支持对文件夹的处理，这里将上面的2张图片放在一个文件夹中，然后使用命令
rembg -p test
rembg 还提供了 web 端的接口，执行 rembg-server 启动后台服务，这里面使用的是 flask 框架
然后在浏览器中输入
http://127.0.0.1:5000/?url=https://image.xugaoxiang.com/imgs/2020/12/bf5f5553a596319b.jpgurl= 后面的是图片的地址，可以看出 rembg 是支持 https 的
代码中调用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代码运行的结果跟上面是一样的
