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使用cx_Oracle调用Oracle存储过程的方法示例
Oct 07 Python
Python GUI编程 文本弹窗的实例
Jun 11 Python
解决python中导入win32com.client出错的问题
Jul 26 Python
Django文件存储 默认存储系统解析
Aug 02 Python
解决python 上传图片限制格式问题
Oct 30 Python
使用python模拟高斯分布例子
Dec 09 Python
解决python 读取 log日志的编码问题
Dec 24 Python
Python 开发工具PyCharm安装教程图文详解(新手必看)
Feb 28 Python
python实现图像全景拼接
Mar 27 Python
python根据完整路径获得盘名/路径名/文件名/文件扩展名的方法
Apr 22 Python
Python self用法详解
Nov 28 Python
python实现简易名片管理系统
Apr 11 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之变量、常量学习笔记
2008/03/27 PHP
ThinkPHP控制器间实现相互调用的方法
2014/10/31 PHP
PHP面向对象中new self()与 new static()的区别浅析
2017/08/17 PHP
WordPress 照片lightbox效果的运用几点
2009/06/22 Javascript
jquery获取table中的某行全部td的内容方法
2013/03/08 Javascript
nodejs教程之制作一个简单的文章发布系统
2014/11/21 NodeJs
jq实现左侧显示图片右侧文字滑动切换效果
2015/08/04 Javascript
尝试动手制作javascript放大镜效果
2015/12/25 Javascript
基于JavaScript实现表单密码的隐藏和显示出来
2016/03/02 Javascript
纯JS代码实现一键分享功能
2016/04/20 Javascript
javascript构造函数以及原型对象的理解
2017/01/13 Javascript
JS 学习总结之正则表达式的懒惰性和贪婪性
2017/07/03 Javascript
jQuery实现简单的计时器功能实例分析
2017/08/29 jQuery
JS库之wow.js使用方法
2017/09/14 Javascript
JS实现商品橱窗特效
2020/01/09 Javascript
详解JavaScript中分解数字的三种方法
2021/01/05 Javascript
[31:01]2014 DOTA2国际邀请赛中国区预选赛5.21 CNB VS Orenda
2014/05/23 DOTA
Python中optionParser模块的使用方法实例教程
2014/08/29 Python
Python自动重试HTTP连接装饰器
2015/04/28 Python
python类继承用法实例分析
2015/05/27 Python
python采集百度百科的方法
2015/06/05 Python
Pandas之排序函数sort_values()的实现
2019/07/09 Python
Python 仅获取响应头, 不获取实体的实例
2019/08/21 Python
Python爬虫:将headers请求头字符串转为字典的方法
2019/08/21 Python
手把手教你用纯css3实现轮播图效果实例
2017/05/04 HTML / CSS
实列教程 一款基于jquery和css3的响应式二级导航菜单
2014/11/13 HTML / CSS
HTML5 与 XHTML2
2008/10/17 HTML / CSS
房地产项目策划书
2014/02/05 职场文书
银行求职自荐信
2014/06/30 职场文书
机关作风建设剖析材料
2014/10/11 职场文书
先进典型事迹材料
2014/12/29 职场文书
2015年国际护士节演讲稿
2015/03/18 职场文书
2015年后勤工作总结范文
2015/04/08 职场文书
通知函的格式
2015/04/27 职场文书
centos8安装nginx1.9.1的详细过程
2021/08/02 Servers
实现一个简单得数据响应系统
2021/11/11 Javascript