Python批量读取多个文件夹下子文件夹的图片-lfw数据集为例

做实验需要用到lfw数据集，网上读取方式太过麻烦，因此自己写了一个简单易懂的代码

import numpy as np #导入numpy模块
from PIL import Image #入PIL模块用于读取图片，也可使用opencv
import os 

if __name__ == "__main__":
    path = 'data/lfw/'	#数据集路径
    #循环遍历lfw数据集下的所有子文件
    for name_file in os.listdir(path):
    #遍历子文件下的所有图片文件
        for img_file in os.listdir(path+name_file):
            print(img_file)	#打印当前读取的图片名
            #以下代码根据需要更改
            img = Image.open(path+name_file+'/'+img_file)	#读取文件
            img = img.resize((224,224))	#归一化，根据需要添加
            img = np.array(img,dtype = np.float64)	#转为float64类型的Numpy数组