python - 'conv2d_2/convolution' 1减3导致的负维度大小

Question

我在 Keras 中声明输入层时收到此错误消息。

ValueError: Negative dimension size caused by subtracting 3 from 1 for 'conv2d_2/convolution' (op: 'Conv2D') with input shapes: [?,1,28,28], [3,3,28,32].

我的代码是这样的

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

示例应用程序:https://github.com/IntellijSys/tensorflow/blob/master/Keras.ipynb

Answer 1

默认情况下，Convolution2D ( https://keras.io/layers/convolutional/ ) 期望输入的格式为 (samples, rows, cols, channels)，即“channels-last”。您的数据似乎采用以下格式(样本、 channel 、行、列)。在声明 Convolution2D 层时，您应该能够使用可选关键字 data_format = 'channels_first' 来解决此问题。

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