CNN-Mnist.py 文件源码

def load_data():  
    data = np.empty((42000,1,28,28),dtype="float32")   #empty?ones????????????????????  
    label = np.empty((42000,),dtype="uint8")  

    imgs = os.listdir("./mnist")   #?????????  
    num = len(imgs)  
    for i in range(num):  
        img = Image.open("./mnist/"+imgs[i])   #???????Image????  
        arr = np.asarray(img,dtype="float32")  #?img?????????  
        data[i,:,:,:] = arr   #?????????data  
        label[i] = int(imgs[i].split('.')[0])   #?????????????  
    return data,label