Lesen von Bildern in 4D Numpy Array mit Dimension (num_img, channel, dim1, dim2)

Question

Ich habe einen Satz von 1000 Graustufenbildern (28x28), ich möchte sie in 4D numpy array (Anzahl der Bilder, 1 , img_dim1, img_dim2). Das Folgende ist mein Code, aber es funktioniert nicht richtig. Irgendeine Idee, wie ich das Problem im Code beheben kann?

from PIL import Image 
import numpy as np 
import os 

mypath=os.path.dirname('path/to/directory/') 
def load_dataset() : 
    data =np.zeros((1000,1,28,28), dtype=np.float64) 
    for fname in os.listdir(mypath): 
     pathname = os.path.join(mypath, fname) 
     img = Image.open(pathname) 
     data = np.dstack((data, img)) 
    return data 
data=load_dataset() 
print(data.shape) 

Answer 1

Das Problem durch die Verwendung append und das Hinzufügen einer neuen Achse np.newaxis Arbeits gelöst wurde

from PIL import Image 
import numpy as np 
import os 

mypath=os.path.dirname('path/to/directory/') 
def load_dataset() : 
    data =[] 
    for fname in os.listdir(mypath): 
     pathname = os.path.join(mypath, fname) 
     img = Image.open(pathname) 
     img1 = img[np.newaxis,:,:] 
     data.append(img1) 
    return data 

data= load_dataset() 
data_x=np.array(data) 
print data_x.shape