Python用5行代码实现批量抠图的示例代码


Posted in Python onApril 14, 2020

前言

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

效果展示

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

Python用5行代码实现批量抠图的示例代码

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

Python用5行代码实现批量抠图的示例代码

因为本身是PNG图片,而且原图是白色背景,所以看不出什么区别。为了显示效果,我把原图和抠好的图放到一张黄色背景图片上:

Python用5行代码实现批量抠图的示例代码

这样一看效果明显多了,感觉抠图效果还是非常好的。但是吧,抠这种简单的图片,不怎么过瘾,我们再来看看复杂一点的图片:

Python用5行代码实现批量抠图的示例代码

这张图片背景色比之前复杂一些,而且有渐变,我们来看看抠图后的效果如何:

Python用5行代码实现批量抠图的示例代码

这个原图背景不是白色,我就不弄黄色背景了,赶紧这个效果也还算满意,那么多人物的图片呢,我们再看看下面这张图片:

Python用5行代码实现批量抠图的示例代码

这里有三个人,我们看看程序能不能自动抠出来:

Python用5行代码实现批量抠图的示例代码

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

Python用5行代码实现批量抠图的示例代码

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

Python用5行代码实现批量抠图的示例代码

哈哈,不仅识别出了人,还把火炬识别出来并抠了出来。总的来说,在完成人物抠图方面还是没有什么问题的。

环境

看完效果,那么应该怎么实现呢?这就需要用到paddlepaddle了,那paddlepaddle是什么呢?paddlepaddle是一个开源的深度学习工具,我们使用该工具可以仅用十几行代码实现迁移学习。在使用之前,我们先来安装paddlepaddle,我们可以进入官网https://www.paddlepaddle.org.cn/,官网上用安装指引。为了方便,这里直接使用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:/CodeField/Workplace/PythonWorkplace/PillowTest/11_yellow/img/'
# 获取目录下的文件
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:/CodeField/Workplace/PythonWorkplace/PillowTest/11_yellow/img/'	# 文件目录
files = [path + i for i in os.listdir(path)]	# 获取文件列表
results = humanseg.segmentation(data={'image':files})	# 抠图

到此这篇关于Python用5行代码实现批量抠图的示例代码的文章就介绍到这了,更多相关Python 批量抠图内容请搜索三水点靠木以前的文章或继续浏览下面的相关文章希望大家以后多多支持三水点靠木!

Python 相关文章推荐
Python深入学习之内存管理
Aug 31 Python
Python实现抓取页面上链接的简单爬虫分享
Jan 21 Python
python文件特定行插入和替换实例详解
Jul 12 Python
简单实现python进度条脚本
Dec 18 Python
python爬虫增加访问量的方法
Aug 22 Python
django框架中ajax的使用及避开CSRF 验证的方式详解
Dec 11 Python
Python实现大数据收集至excel的思路详解
Jan 03 Python
python新式类和经典类的区别实例分析
Mar 23 Python
Django 实现对已存在的model进行更改
Mar 28 Python
python如何代码集体右移
Jul 20 Python
Python代码覆盖率统计工具coverage.py用法详解
Nov 25 Python
基于Python 函数和方法的区别说明
Mar 24 Python
在jupyter notebook中调用.ipynb文件方式
Apr 14 #Python
使用jupyter notebook将文件保存为Markdown,HTML等文件格式
Apr 14 #Python
Python使用pyyaml模块处理yaml数据
Apr 14 #Python
Jupyter Notebook打开任意文件夹操作
Apr 14 #Python
Python requests模块cookie实例解析
Apr 14 #Python
Python requests模块session代码实例
Apr 14 #Python
使用Jupyter notebooks上传文件夹或大量数据到服务器
Apr 14 #Python
You might like
PHP 5.0对象模型深度探索之绑定
2006/09/05 PHP
PHP生成网页快照 不用COM不用扩展.
2010/02/11 PHP
如何使用php判断所处服务器操作系统的类型
2013/06/20 PHP
php递归json类实例
2014/12/02 PHP
php面向对象中static静态属性和静态方法的调用
2015/02/08 PHP
用javascript实现点击链接弹出"图片另存为"而不是直接打开
2007/08/15 Javascript
一些技巧性实用js代码小结
2009/10/14 Javascript
JQuery 学习技巧总结
2010/05/21 Javascript
jquery读取xml文件实现省市县三级联动的方法
2015/05/29 Javascript
AngularJS学习笔记之ng-options指令
2015/06/16 Javascript
jQuery解决浏览器兼容性问题案例分析
2016/04/15 Javascript
jquery实现无刷新验证码的简单实例
2016/05/19 Javascript
xtemplate node.js 的使用方法实例解析
2016/08/22 Javascript
JS动态加载脚本并执行回调操作
2016/08/24 Javascript
详解js的延迟对象、跨域、模板引擎、弹出层、AJAX【附实例下载】
2016/12/19 Javascript
js实现网页的两个input标签内的数值加减(示例代码)
2017/08/15 Javascript
javascript 面向对象实战思想分享
2017/09/07 Javascript
AngularJS实现自定义指令及指令配置项的方法
2017/11/20 Javascript
vue.js使用v-model指令实现的数据双向绑定功能示例
2018/05/22 Javascript
详解小程序中h5页面onShow实现及跨页面通信方案
2019/05/30 Javascript
修改NPM全局模式的默认安装路径的方法
2020/12/15 Javascript
python网络编程学习笔记(一)
2014/06/09 Python
python线程池threadpool使用篇
2018/04/27 Python
用python标准库difflib比较两份文件的异同详解
2018/11/16 Python
在线服装零售商:SheIn
2016/07/22 全球购物
加拿大租车网站:Enterprise Rent-A-Car
2018/07/26 全球购物
土木工程毕业生自荐信
2013/09/21 职场文书
后勤园长自我鉴定
2013/10/17 职场文书
计算机专业个人简短的自我评价
2013/10/23 职场文书
学习2014年全国两会心得体会
2014/03/12 职场文书
政府采购方案
2014/06/12 职场文书
学生违纪检讨书200字
2014/10/21 职场文书
2014年学校教学工作总结
2014/12/06 职场文书
初三英语教学反思
2016/02/15 职场文书
springboot集成flyway自动创表的详细配置
2021/06/26 Java/Android
yyds什么意思?90后已经听不懂00后讲话了……
2022/02/03 杂记