Python numpy Valueerror wenn I-Bilddaten zu laden und sie auf einen Array

Question

I nicht die solution.My Bildform ist 128 * 128 * 3 finden konnte, hat es drei Kanäle, aber es verursacht auch den Fehler

File "E:/ML/keras_test/vgg.py", line 30, in load_data data[i,:,:,:] = arr

ValueError: could not broadcast input array from shape (128,128) into shape (128,128,3)

Mein Code unten ein:

def load_data(path): 
data = np.empty((12755,128,128,3),dtype="uint8") 
label = np.empty((12755,),dtype="uint8") 


imgs = [] 
imgs_name = [] 
for each_person in os.listdir(path): 
    temp = os.path.join(path,each_person) 
    for each_image in os.listdir(temp): 
     imgs.append(temp + "\\" + each_image) 
     imgs_name.append(each_image) 

num = len(imgs) 
for i in range(num): 
    img = Image.open(imgs[i]) 
    arr = np.asarray(img,dtype="uint8") 
    print arr.shape 
    data[i,:,:,:] = arr 
    label[i] = int(imgs_name[i].split('.')[0]) 

print 'load_data is ok!' + str(data.shape[0]) 
return data,label 

Answer 1

Sie versuchen, die Originaldaten in ein kleines Paket zu setzen, was unmöglich ist. Ich denke, Sie versuchen, ein Bild mit RGB-Kanal in ein Graustufenbild mit einem Kanal zu übertragen. Versuchen

Bezug = (imgs.sum (Achse = 2)/3) .reshape ((12755, -1))

Die resultierenden datum ist ein 12755 x 16384 Array.