python批量读取文件名_python批量读取文件夹的图片并处理成模型输入格式

我们知道在训练图像模型时，需要从文件夹里读取图片。

这里演示下，怎样从文件夹里批量读取图片，并处理成我们想要的

x,y 即特征和标签。

输入维度理解

以卷积神经网络为例子。

我们知道卷积神经网络conv2d输入数据的形状为(batch_size，height，width，depth)，其中第一维表示图像的batch大小,我们常设定为None，其他三个维表示图像的各个属性，即高度，宽度和深度。深度就是色彩通道的数量。例如，RGB图像的深度为3，而灰度图像的深度为1。

通俗理解。一张图片。我们输入的格式为（高，宽，通道）

即我们需要把图片处理成相同的高宽。

以下面这个文件夹为例，图片大小不一样，有的12kB,有的不是，高宽肯定不一样。

#python+opencv读取文件图片并处理

import os

import numpy as np

import cv2

cat=[]#用于存储x。

def ReFileName(dirPath):

"""

:param dirPath: 文件夹路径

:return:

"""

# 对目录下的文件进行遍历

for file in os.listdir(dirPath):

# 判断是否是文件

if os.path.isfile(os.path.join(dirPath, file)) == True:

c= os.path.basename(file)

name = dirPath + '\\' + c

img = cv2.imread(name)

img=cv2.resize(img,(100,100))#使尺寸大小一样

cat.append(img)

dirPath = r"C:\Users\Shineion\Desktop\cat"#文件路径

ReFileName(dirPath)#调用函数

cat=np.array(cat)#数据

print('输入x.shape',cat.shape)

biaoqian=[1 for i in range(len(cat))]#标签处理

biaoqian=np.array(biaoqian)#标签

print('输入y.shape',biaoqian.shape)

得到的是我们想要的x,y格式

.说明

注意：cv2.imread(name) 读取的图片绝对路径name ，不能含有中文。中文报错。

1.代码里只读取啦一种类别的图片：猫,并将猫的标签设置为数字1。

一般我们会将不同类别的图片放在不同文件夹。读取另外类型图片就是重复上述步骤，并修改标签数字。

2.如果是多种类型图片放在同一个文件夹里，可以根据文件名来来选择图片，并存储到不同数组（如代码中的cat）。选择文件名一般是根据正则。

正则链接：

python :re模块基本用法

可能有帮助的链接：

python 批量修改图片名（也可以用来批量修改文件名）

3.代码里只读取图片，并未对图片进行任何处理。如识别猫狗，肯定要消除背景 和颜色影响的。opencv里有很多方法，参考python3+opencv学习笔记汇总目录（适合基础入门学习）。

处理代码直接加在 img=cv2.resize(img,(100,100))#使尺寸大小一样 后面就可以。

不同任务处理方法不一样。

4.其他类型的输入数据格式。不一定都是(batch_size，height，width，depth)。

如一维输入

自己最后改变数据维度就可以。

电气专业的计算机小白，写博文不容易。如果你觉得本文对你有用，请点个赞支持下。谢谢。