深度学习中彩色图片批量处理成灰度图片(Python)

在进行深度学习之前,往往需要创建一个用于训练的图片数据集,且一般要求训练集具有统一 的格式和大小。本文以批量处理大小不一 的jpg图片为例,介绍批量处理的一些问题。方法仅供参考。
程序如下:

###本程序用于将.jpg格式的图片批量转化为.npg灰度图像
import os
from PIL import Image

infile = './flowers/'
outfile = './flowers-gray/'
folddirs = os.listdir(infile)
filecount = 0
for foldername in folddirs:
    filepath = infile+ foldername

filedirs = os.listdir(filepath)
img_num = int(len(filedirs))
for filename in filedirs:
    img_path = filepath + '/' + filename
    img = Image.open(img_path).convert('L')
    out = img.resize((64,64),Image.ANTIALIAS)
    filecount = filecount + 1

    print(filecount)

    newfilename = outfile + foldername + '/' + foldername + str('_')+str(filecount) + '.png'
    out.save(newfilename)

特别说明:

infile = './flowers/'   #用于存储原始图片
outfile = './flowers-gray/'  #用于存储灰度图片

本文数据下载地址:
http://download.tensorflow.org/example_images/flower_photos.tgz
解压后的文件夹包含了5个子文件夹,每一个子文件夹的名称为一种花的名称,代表了不同的类别,每种花大概有几百张图片,每一张图片都是RGB色彩模式的,大小也不相同。若官网下载太慢或者无法下载,可在本博客留言索取。
为方便阅读程序,本程序需要手动创建文件夹。
文件夹如下:
在这里插入图片描述
flowers中的文件夹,flowers-gray下文件夹一样:
在这里插入图片描述

END!

  • 2
    点赞
  • 24
    收藏
    觉得还不错? 一键收藏
  • 5
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 5
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值