做实验需要用到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数组