最新下载
热门教程
- 1
- 2
- 3
- 4
- 5
- 6
- 7
- 8
- 9
- 10
Python用5行代码如何实现批量抠图 Python用5行代码实现批量抠图方法
时间:2020-04-14 编辑:袖梨 来源:一聚教程网
本篇文章小编给大家分享一下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:Userszaxwz>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:CodeFieldWorkplacePythonWorkplacePillowTest11_yellow>python 抠图.py
[2020-03-10 21:42:34,587] [ INFO] - Instalpng deeplabv3p_xception65_humanseg module
[2020-03-10 21:42:34,605] [ INFO] - Module deeplabv3p_xception65_humanseg already installed in C:Userszaxwz.paddlehubmodulesdeeplabv3p_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}) # 抠图
相关文章
- Golang ProtoBuf的基本语法详解 10-20
- Python识别MySQL中的冗余索引解析 10-20
- Python+Pygame绘制小球代码展示 10-18
- Python中的数据精度问题介绍 10-18
- Python随机值生成的常用方法介绍 10-18
- python3解压缩.gz文件分析 09-27