python - ValueError:检查目标时出错:预期(keras 序列模型层)具有 n 维，但得到具有形状的数组

Question

我已经加载了图像来训练我的模型识别这些图像中的一个特征。

Xtrain 是一个形状为 (1380,200,200,3 ) 的 numpy ndarray包含 1380 张大小为 200 x 200 像素的 RGB 格式图像

Ytrain有目标。形状 (1380,2)

当我训练我的模型 ( model.fit(Xtrain,Ytrain) ) 时，我似乎在每个层上都出现了值错误。好像输入都是 Xtrain然后 Ytrain ...

ValueError: Error when checking target: expected batch_normalization_24 to have 4 dimensions, but got array with shape (1380, 2)

图片:

Answer 1

Keras 的批量归一化层的输出形状与其输入相同。由于您只有两个标签，因此序列模型中的最后一层应生成两个输出。可以考虑加一个Dense层如:

model.add(Dense(2), activation='relu')

我还建议使用 print(model.summary()) 检查模型的架构并确保输入和输出与您的数据集匹配，反之亦然。