python读取多个文件夹/子文件夹里面的图片，并且将图片转为灰度图

最近在做CNN，因此要大量读取图片，朋友给了这个代码给我，我觉得不错就分享一下。用了这个代码一下子就能读取所有子文件夹下所有图片了

这是dataset，我把图片全部存放在他的子文件夹里，便于区分各类图片。

我的图片全部存放在子文件夹中：

然而当我想在python直接用image函数读取所有图片时候，报错了emmm
无奈直接当伸手党拿了一份代码，顺便与大家分享

#skimage.io.imread()
#skimage.color.rgb2gray()
import os
import skimage.io
import skimage.color
data_dir = 'dataset/'#文件地址/名称
classes = os.listdir(data_dir)
data = []
for cls in classes:
    files = os.listdir(data_dir+cls)
    for f in files:        
        img = skimage.io.imread(data_dir+cls+"/"+f)
        img = skimage.color.rgb2gray(img)#将图片转为灰度图
        data.append({
            'x':img,
            'y':cls
        })
        
random.shuffle(data)