[Python知识库]用python抠图

GitHub上有个好完的项目，rembg

简单实用的删除图像背景/抠图工具
能把图片背景去掉，抠出来主要的内容。

如何使用？

1、安装

CPU support:

pip install rembg

GPU support:

pip install rembg[gpu]

安装的东西挺多的，安装完成后就可以使用了

2、使用

使用方式很多；可以敲一行命令直接抠图、也可以部署到服务器上用浏览器抠图、也可以写py代码抠图、也可以用docker等等。

2.1 以命令行方式使用

2.1.1、在命令行中，切换到使用的虚拟环境，我的虚拟环境起名叫py310

conda activate py310

2.1.2、开始抠图

# 1、扣本地的图
rembg i path/to/input.png path/to/output.png
# 2、扣网上的图
curl -s http://input.png | rembg i > output.png
# 3、扣整个文件夹的图
rembg p path/to/input path/to/output

抠图前：（图片来自百度）

抠图后：

2.2以服务的方式使用

也就是可以部署到自己云服务器上，整个网页或者app什么的，就可以随时随地抠图了。

2.2.1 官方网页版

# 在虚拟环境中，敲以下命令：
rembg s

会提示服务地址：http://0.0.0.0:5000
注意这个地址不是直接打开的，后边得加个docs，也就是：http://0.0.0.0:5000/docs
get方式适用于网上的URL地址
post方式适合文件流，这里我们使用post的方式

点Try it out

上传图片

点execute

得到抠图结果：

2.2.2 自己做网页用

新建一个html文件，添加如下代码：

<form
    action="http://http://0.0.0.0:5000"
    method="post"
    enctype="multipart/form-data"
>
    <input type="file" name="file" />
    <input type="submit" value="upload" />
</form>

打开html文件

在浏览器中上传文件之后，会自动抠图

2.3以一个python库的方式使用

2.3.1 以文件流的方式

Input and output as bytes

from rembg import remove

input_path = 'input.png'
output_path = 'output.png'

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

2.3.2 用PIL图像的方式

Input and output as a PIL image

from rembg import remove
from PIL import Image

input_path = 'input.png'
output_path = 'output.png'

input = Image.open(input_path)
output = remove(input)
output.save(output_path)

2.3.3 甚至于用numpy array

Input and output as a numpy array

from rembg import remove
import cv2

input_path = 'input.png'
output_path = 'output.png'

input = cv2.imread(input_path)
output = remove(input)
cv2.imwrite(output_path, output)

2.4以docker的方式使用（这个以后试试）

docker run -p 5000:5000 danielgatis/rembg s

写了这么多，问一个灵魂问题：
作为一只程序猿，抠图有什么用呢？我自己一般用来做ppt的时候用。不过好像很多软件都自带了抠图功能，所以rembg这个项目大家感兴趣玩玩就行了。

举个例子：
图1是没抠图插入ppt的样子；图2是抠图之后插入ppt之后的样子。图2对原版ppt遮挡较小，并且没什么违和感。