python - 卷积层和输入数据上的 Keras 错误

Question

我从 CIFAR-100 中过滤了训练和测试数据，我只选择水果和蔬菜父类(super class)。现在，我有 2,500 个训练数据和 500 个测试数据。但是，我收到一个错误，表示卷积层的尺寸输入错误。

我的数组数据形式:

希望有人能帮我解决这个问题，谢谢。

Answer 1

您的输入数据应具有形状(2500,3,32,32)，似乎您在预处理步骤中丢失了两个维度，请修复这些维度或将数据重新整形为:

inputData = inputData.reshape((2500, 3, 32, 32)).

一般来说，卷积层的输入是(numSamples、numChannels、width、height)。请注意，使用 tensorflow 后端时， channel 数维度位于末尾。