大佬教你，Python用5行代码实现批量抠图

最新推荐文章于 2022-03-24 11:10:49 发布

BC_COM

最新推荐文章于 2022-03-24 11:10:49 发布

阅读量1.3k

点赞数 3

分类专栏： python

本文链接：https://blog.csdn.net/BC_COM/article/details/105731617

版权

python 专栏收录该内容

16 篇文章 7 订阅

订阅专栏

对于会PhotoShop的人来说，抠图是非常简单的操作了，有时候几秒钟就能扣好一张图。不过一些比较复杂的图，有时候还是要画点时间的，今天就给大家带了一个非常快速简单的办法，用Python来批量抠取人像。

效果展示

开始吧，我也不看好什么自动抠图，总觉得不够精确，抠不出满意的图。下面我就直接展示一下效果图吧。我们先看看原图

这张图片背景未纯色，我们平时用PhotoShop抠起来也比较简单，对我们计算机来说也不是什么难题，下面是效果图：

因为本身是PNG图片，而且原图是白色背景，所以看不出什么区别。再来一张：

这样一看效果明显多了，感觉抠图效果还是非常好的

我们再看看下面这张篮球妹子图片：

虽然是有点瑕疵，不过还是很不错了，下面我们看看最后一个例子：

这个就比前面的都复杂的多，那么效果如何呢，我们来看看：

哈哈，效果棒不棒。总的来说，在完成人物抠图方面还是没有什么问题的。

环境
为了方便，这里直接使用pip安装CPU版本的。
我们执行下列语句：

python -m pip install paddlepaddle -i https://mirror.baidu.com/pypi/simple

安装完成后我们可以在环境中测试一下是否成功。我这里使用命令行窗口，先运行python.exe（前提是你已经配置了环境变量）

C:\Users\zaxwz>python

然后在程序中运行如下代码：

Python 3.7.6 (tags/v3.7.6:43364a7ae0, Dec 19 2019, 00:42:30) [MSC v.1916 64 bit (AMD64)] on win32
Type "help", "copyright", "credits" or "license" for more information.
>>> import paddle.fluid
>>> paddle.fluid.install_check.run_check()

如果控制台显示Your Paddle is installed successfully! Let’s start deep Learning with Paddle now就代表我们已经安装成功了。另外我们还需要安装paddlehub：

pip install -i https://mirror.baidu.com/pypi/simple paddlehub

下面我们就可以开始写代码了。

实现抠图

实现抠图的代码很简单，大概分为下面几个步骤：

导入模块
加载模型
获取文件列表
抠图

实现起来没有什么难度，为了方便读代码，我将代码写清楚一点：

# 1、导入模块
import os
import paddlehub as hub
# 2、加载模型
humanseg = hub.Module(name='deeplabv3p_xception65_humanseg')
# 3、获取文件列表
# 图片文件的目录
path = 'D:PillowTest/11_yellow/images/'
# 获取目录下的文件
files = os.listdir(path)
# 用来装图片的
imgs = []
# 拼接图片路径
for i in files:
    imgs.append(path + i)
#抠图
results = humanseg.segmentation(data={'image':imgs})

我们在控制台运行一下这个程序：

D:\CodeField\Workplace\PythonWorkplace\PillowTest\11_yellow>python 抠图.py
[2020-03-10 21:42:34,587] [    INFO] - Installing deeplabv3p_xception65_humanseg module
[2020-03-10 21:42:34,605] [    INFO] - Module deeplabv3p_xception65_humanseg already installed in C:\Users\zaxwz\.paddlehub\modules\deeplabv3p_xception65_humanseg
[2020-03-10 21:42:35,472] [    INFO] - 0 pretrained paramaters loaded by PaddleHub

运行完成后，我们可以在项目下看到humanseg_output目录，抠好的图片就会存放在该目录下。当然了，上面的代码我们在获取文件列表的操作还可以简化一下：

import os, paddlehub as hub
humanseg = hub.Module(name='deeplabv3p_xception65_humanseg')        # 加载模型
path = 'D:/PillowTest/11_yellow/img/'    # 文件目录
files = [path + i for i in os.listdir(path)]    # 获取文件列表
results = humanseg.segmentation(data={'image':files})    # 抠图

好，这就是最终的，喜欢的可以关注哦