ValueError：输入到`.fit（）`应该具有等级4。具有形状的数组

Question

我遇到了这个错误，但不知道为什么。

Traceback (most recent call last):
  File "C:\Users\JoshG\PycharmProjects\LeNet\LeNet.py", line 134, in <module>
    train_generator.fit(xtrain.reshape(704, 227, 227))
ValueError: cannot reshape array of size 36070300 into shape (704,227,227)

以下是导致错误消息的行。

#Import the packages
from keras.preprocessing.image import ImageDataGenerator

# Image Data Augmentation
train_generator = ImageDataGenerator(rotation_range=2, horizontal_flip=True, zoom_range=.1)
val_generator = ImageDataGenerator(rotation_range=2, horizontal_flip=True, zoom_range=.1)
test_generator = ImageDataGenerator(rotation_range=2, horizontal_flip=True, zoom_range=.1)

# Fitting the augmentation defined above to the data
train_generator.fit(xtrain.reshape(704, 227, 227))
val_generator.fit(x_val.reshape(704, 227, 227))
test_generator.fit(xtest.reshape(704, 227, 227))

更新： 我将代码修改为以下内容：

# Fitting the augmentation defined above to the data
train_generator.fit(xtrain.reshape(704, 227, 227))
val_generator.fit(x_val.reshape(704, 227, 227))
test_generator.fit(xtest.reshape(704, 227, 227))

并且现在遇到此错误：

Traceback (most recent call last):
  File "C:\Users\JoshG\PycharmProjects\LeNet\LeNet.py", line 136, in <module>
    val_generator.fit(x_val.reshape(704, 227, 227))
ValueError: cannot reshape array of size 4946784 into shape (704,227,227)

因此，我删除了704，并留下-1，如下所示，

# Fitting the augmentation defined above to the data
train_generator.fit(xtrain.reshape(-1, 227, 227))
val_generator.fit(x_val.reshape(-1, 227, 227))
test_generator.fit(xtest.reshape(-1, 227, 227))

但是我收到此错误消息：

Traceback (most recent call last):
  File "C:\Users\JoshG\PycharmProjects\LeNet\LeNet.py", line 135, in <module>
    train_generator.fit(xtrain.reshape(-1, 227, 227))
  File "C:\Users\JoshG\AppData\Local\Programs\Python\Python39\lib\site-packages\keras_preprocessing\image\image_data_generator.py", line 935, in fit
    raise ValueError('Input to `.fit()` should have rank 4. '
ValueError: Input to `.fit()` should have rank 4. Got array with shape: (704, 227, 227)