装逼神器，5 行 Python 代码 实现一键批量扣图，你get到了吗

在日常生活或者工作中，经常会遇到想将某张照片中的人物抠出来，然后拼接到其他图片上去。专业点的人可以使用 PhotoShop 的“魔棒”工具进行抠图，非专业人士则使用各种美图 APP 来实现，但是这两类方式毕竟处理能力有限，一次只能处理一张图片，而且比较复杂的图像可能耗时较久。那今天就来向大家展示第三种扣图方式——用 Python代码来实现 一键批量抠图。

1. 准备工作- 安装paddlepaddle

既然要装逼，准备工作是少不了的。所谓“站在巨人的肩膀上，干起活来事半功倍”，我们这里的“巨人”就是 paddlepaddle 了，中文名称叫 “飞桨” ，那么这个 paddlepaddle 是什么呢？

它是“ 源于产业实践的开源深度学习平台 ，致力于让深度学习技术的创新与应用更简单”，直白点就是我帮你实现了深度学习底层框架，你只要有创意就可以在我平台上运用少量简单代码轻松实现。它的官网是：https://www.paddlepaddle.org.cn/ 。

它的安装比较简单，官网首页就有安装指引，可以通过「安装」菜单，查找到各个系统安装详细及注意事项，如下图所示，我们这里根据官网的安装指引，使用 pip 方式来安装 CPU 版本。

 

本文以MacOS系统为例：

我们首先执行以下命令安装（推荐使用百度源）：：

或者：

从安装过程中，可以看到在安装paddlepaddle库时，需要安装如下依赖库：

安装成功后，我们在 python 环境中测试一下是否安装成功（这个也是按照官网指引来做），我们切换到 python 环境，运行如下代码：

如果能看到 Your Paddle Fluid is installed successfully 就表示安装成功了。

2. 准备工作- 安装paddlehub

要实现本文的一键批量扣图需求，需要借助PaddleHub人像分割模型来实现。

PaddleHub 是基于 PaddlePaddle 开发的预训练模型管理工具，可以借助预训练模型更便捷地开展迁移学习工作，目前的预训练模型涵盖了图像分类、目标检测、词法分析、语义模型、情感分析、视频分类、图像生成、图像分割、文本审核、关键点检测等主流模型。

PaddleHub官网：

PaddleHub项目地址：

更多PaddleHub预训练模型教程合集课程可见：

介绍完了项目，接下来我们开始在线安装 paddlehub ：

或者按指定版本安装：

安装完成后，我们就可以开始运用了。

3. 一键扣图代码实现

我们的实现步骤很简单：

• 导入模块
• 加载模型
• 获取图片文件
• 调用模块抠图

其中扣图功能主要采用PaddleHub DeepLabv3+模型deeplabv3p_xception65_humanseg。

下面我们看具体扣图代码实现(demo.py)：

示例中，我将图片放在代码文件夹的同级目录 images文件夹下，运行代码后，输出的抠图图片会自动放在代码同级目录的 humanseg_output 目录下，文件名称跟原图片的名称相同，但是文件格式是 png 。

其中示例 images 目录下放了9张图片，为了兼顾不同读者喜好的口味，示例图片中既包括了帅哥，也有美女哦，并且将他们缩略图放大了，如下：

 

原图片

运行程序后，上述示例代码运行结果如下所示。

 

运行成功后，在 humanseg_output 目录下生成了9张图片，同样的，扣图的结果如下所示：

 

抠图图片

我们可以看到程序将每张图片中的人物（可以是一个人，也可以是多个人）识别出来，并且抠出来成图，背景是白色。虽然有些细节处还有些许瑕疵，但是看起来还算不错。

4. 需要注意的坑

在运行示例代码时，如果没有单独安装模型deeplabv3p_xception65_humanseg，默认会自动在执行前进行安装。但安装完成后，执行结果并没有生成扣图结果及humanseg_output目录，输出结果类似如下所示：

 

正常情况下，在生成扣图数据，打印results时，应该是类似如下结构才对：

 

可以通过单独安装模型并指定安装版本来解决。

具体原因没有细究，默认自动安装模型时，版本为1.2.0，猜测由于还是模型版本不兼容问题导致。如果大家对Python感兴趣的话，可以加一下我的微信哦：abb436574，免费领取一套学习资料和视频课程哟~

5. 总结

本文基于 paddlepaddle 平台，利用PaddleHub DeepLabv3+模型(deeplabv3p_xception65_humanseg)，使用简单的五行代码就实现了批量抠图。有些读者可能会想，上述示例中提供的代码行数不止五行代码吧，在上述示例中，真正实现扣图的主代码其实只需要下面五行：

利用PaddleHub DeepLabv3+模型 不仅可以实现一键扣图，还可以进行图片合成，视频合成等。利用好它不仅解放了人的双手和双眼，而且为某些程序猿/程序媛的装逼工具箱提供了一件宝器。下次如果碰到某个女生或者闺蜜在为抠图发愁，别忘了掏出神器，赢得芳心哦！