python - Convolution2D 在 Keras 中给出错误的输出形状

Question

我正在关注 here 的教程并坚持简单的模型创建

model = Sequential()
model.add(Convolution2D(32, kernel_size=(3, 3), activation='relu', input_shape=(1, 28, 28)))
model.output_shape

这应该给出 (None, 32, 26, 26) 输出，但我得到 (None, -1, 26, 32)。我猜想提供的参数有问题或者我需要进行额外的 reshape 。也许 Keras 中的 Convolution2D 和 Conv2D 发生了一些变化？我正在使用 Theano 后端运行 2.2.4

Answer 1

channel 似乎位于输入数据的第一个轴中。因此，您需要将 data_format 参数设置为 "channels_first":

model = Sequential()
model.add(Convolution2D(32, kernel_size=(3, 3),
                        data_format='channels_first',
                        activation='relu', input_shape=(1, 28, 28)))
model.output_shape

这给出:

(None, 32, 26, 26)