python实现图片去背景

rembg 是一个去除图片背景的工具，一行代码轻松地就能去掉图片的背景，那可不比手动抠图快得多嘛。

想想用 PS 的经历，一点一点地把要抠的内容圈出来，这个过程老费时间了。

敲代码多炫酷，敲两行代码就能搞定的事儿，干嘛还得在操作界面上点点点… …

安装第三方库

这里大家要注意一下，rembg 支持的 python 版本是 3.9，如果版本不对应需要调整一下。

并且只能使用 python3.9。

我安装的 python 版本是 3.9.5，使用 rembg 没有任何问题，大家可以参考一下我的版本。

安装命令如下：

#CPU支持
pip install rembg
#GPU支持
pip install rembg[gpu]

果然，还是踩坑了，安装的过程出现问题了。

其实错误信息不看也罢，最后排查出问题的根源，完全就是因为网络导致的，没有代码层面的事儿。

只要你把网络调整一下，能够稳定地使用谷歌搜索，基本就能正常安装了。

去除单张图片背景

命令行模式

安装完成可以直接在命令行中进行调用，连 python 代码都不用写。

参考命令如下：

rembg i 原图全路径 输出图片全路径
rembg i D:/img/111.jpeg D:/img/aaa.jpeg 

代码模式

同样的功能，如果想要在 python 代码中实现，可参考下面这段代码。

大家自己尝试的时候，只需要修改一下图片的路径就可以了。

from rembg import remove
# 输入图片路径
input_path = 'D:/img/111.jpeg'
# 输出图片路径
output_path = 'D:/img/aaa.jpeg'

with open(input_path, 'rb') as i:
    with open(output_path, 'wb') as o:
        input = i.read()
        output = remove(input)
        o.write(output)

去除文件夹所有图片背景

上面介绍了单张图片背景的方法，那去除多张图片的背景呢？

作为程序员，处理一条数据和多条数据能有啥区别？套个循环不就能搞定了。

但是 rembg 已经帮我们考虑到这个问题了，完全不需要自己写循环去处理。

同样一条简单的命令就能搞定：

rembg p 输入文件夹 输出文件夹
rembg p d:/img d:/img2

去除网络图片背景

除开这两项基本功能，rembg 还支持给定图片的网络地址，直接把图片去除背景后给我们下载在本地。

同样也只需要一个简单的命令就能搞定，连下载图片的操作都省去了，我就想问一句：6 不 6 ？

命令参考如下：

curl -s 网络图片路径 | rembg i > 存储位置全路径
curl -s https://xxx.com/9f6d15e60674.jpg | rembg i > d:/img/666.png