Python 读取文件夹的文件

def get_files(filename):
    class_train = []
    label_train = []
    for train_class in os.listdir(filename):
         for pic in os.listdir(filename+train_class):
             class_train.append(filename+train_class+'/'+pic)
             label_train.append(train_class)
    temp = np.array([class_train,label_train])
    temp = temp.transpose()
    #shuffle the samples
    np.random.shuffle(temp)
    #after transpose, images is in dimension 0 and label in dimension 1
    image_list = list(temp[:,0])
    label_list = list(temp[:,1])
    label_list = [int(i) for i in label_list]
    #print(label_list)
    return image_list,label_list
TrainData,labels=get_files(path)

import numpy as np
import glob
from skimage import io
from skimage import transform 
#读取图片
path='C:/Users/hsy/Desktop/train/'
def read_img(path):
    cate=[path+x for x in os.listdir(path) if os.path.isdir(path+x)]
    imgs=[]
    labels=[]
    for idx,folder in enumerate(cate):
        for im in glob.glob(folder+'/*.jpg'):
            print('reading the images:%s'%(im))
            img=io.imread(im)
            img=transform.resize(img,(64,64))
            imgs.append(img)
            labels.append(idx)
    return np.asarray(imgs,np.float32),np.asarray(labels,np.int32)
data,label=read_img(path)