面部表情识别数据集时遇到的错误,这是我解决错误的方法。
因为数据集是灰度图,当使用ImageDataGeneratortensorflow 并flow_from_directory生成训练和验证集时,
需要根据数据集/图像将 color_mode 指定为
test_generator = test_datagen.flow_from_directory(
'data/test',
target_size=(48, 48),
batch_size=32,
color_mode="grayscale",
class_mode = 'categorical')
在模型的第一层 Conv2D 中 input_shape 应该是
input_shape = (height, width, 1)
1 因为它是 grayscale.