python 读取文件夹下的图片进行处理

python的os模块中有一个listdir函数可以遍历读取文件夹下的文件。

import os

for filename in os.listdir(r"./file"):              #listdir的参数是文件夹的路径
    print ( filename)                                  #此时的filename是文件夹中文件的名称
    

利用上述代码可以读取文件夹下的所有文件，也包括图片。

找到图片以后，我们利用opencv中的imread函数读取图片，将其存储为一个numpy.narray类型的数组

import cv2                   #导入opencv库
img1 = cv2.imread(filenames,cv2.IMREAD_GRAYSCALE)   #读取图片，第二个参数表示以灰度图像读入
if img1 is None:                   #判断读入的img1是否为空，为空就继续下一轮循环
            continue
res1= cv2.resize(img1,(28,28))              #对图片进行缩放，第一个参数是读入的图片，第二个是制定的缩放大小
res1_1 = res1.reshape(1,784)/255       #将表示图片的二维矩阵转换成一维
res1_1_1 = res1_1.tolist()                     #将numpy.narray类型的矩阵转换成list
train_set_x.append(res1_1_1)               #将list添加到已有的list中