我正在建立用于二进制分类的分类器,当我尝试将图像重塑为模型的正确输入类型时,我会遇到此错误。
ValueError: cannot reshape array of size 25000 into shape (50,50,3) ----- BGR
但是当我重塑GrayScale的图像时,它工作得很好。
X=np.array(x).reshape(-1,50,50,1)
这是我加载图像并调整其大小的方法:
training_data=[]
labels=[]
for i in os.listdir():
label=i.split('.')[0]
if label=='cat':
labels.append(0)
else:
labels.append(1)
image=cv2.imread(i)
image_array=cv2.resize(image,(50,50))
training_data.append([image,labels])