ValueError：無法將輸入數組從形狀（22500,3）廣播到形狀（1）

Question

我依靠這里提到的代碼，但進行了少量修改。 我的版本如下：

import numpy as np
import _pickle as cPickle
from PIL import Image
import sys,os

pixels = []
labels = []
traindata = []
data=[]

directory = 'C:\\Users\\abc\\Desktop\\Testing\\images'
for root, dirs, files in os.walk(directory):
    for file in files:
        floc = file
        im = Image.open(str(directory) + '\\' + floc)
        pix = np.array(im.getdata())
        pixels.append(pix)
        labels.append(1)

pixels = np.array(pixels)
labels = np.array(labels)
traindata.append(pixels)
traindata.append(labels)
traindata = np.array(traindata)
# do the same for validation and test data
# put all data and labels into 'data' array
cPickle.dump(data,open('data.pkl','wb'))

運行代碼時，我得到以下信息：

Traceback (most recent call last):
  File "pickle_data.py", line 24, in <module>
    traindata=np.array(traindata)
ValueError: could not broadcast input array from shape (22500,3) into shape (1)

我該如何解決這個問題？

謝謝。

Answer 1

要了解的結構traindata ，我換成你pixels.append(pix)與pixels.append(pix[np.ix_([1,2,3],[0,1,2])])有一些玩具例子。 然后我得到traindata是

[array([[[16, 13, 15],
         [16, 13, 15],
         [16, 13, 15]]]), array([1])]

當您嘗試將traindata轉換為numpy數組時，會出現錯誤，因為它由不同大小的子數組組成。 您可以將每個子數組保存在單獨的numpy數組中，也可以這樣進行：

traindata = np.array([traindata[0][0],traindata[1]], dtype=object)

通過使用dtype=object ，您可以創建由不同大小的元素組成的numpy數組。