Keras:ValueError:输入0与卷积层convolution2d_11不兼容:期望的ndim = 4,而发现ndim = 2。

3
    model.add(Convolution2D(64, 3, 3, border_mode='same', input_shape=(32, 32, 3)))
    model.add(Activation('relu'))
    model.add(Convolution2D(32, 3, 3))
    model.add(Activation('relu'))
    model.add(MaxPooling2D(pool_size=(2, 2)))
    model.add(Dropout(0.25))

    model.add(Convolution2D(64, 3, 3, border_mode='same'))
    model.add(Activation('relu'))
    model.add(Convolution2D(64, 3, 3))
    model.add(Activation('relu'))
    model.add(MaxPooling2D(pool_size=(2, 2)))
    model.add(Dropout(0.25))

    model.add(Flatten())
    model.add(Dense(512))
    model.add(Activation('relu'))
    model.add(Dropout(0.5))
    model.add(Dense(10))
    model.add(Activation('softmax'))

这是错误信息

ValueError                                Traceback (most recent call last)
<ipython-input-21-a60216c72b54> in <module>()
----> 1 model.add(Convolution2D(64, 3, 3, border_mode='same', input_shape=(3, 32, 32)))
      2 model.add(Activation('relu'))
      3 model.add(Convolution2D(32, 3, 3))
      4 model.add(Activation('relu'))
      5 model.add(MaxPooling2D(pool_size=(2, 2)))

/home/pranshu_44/anaconda3/lib/python3.5/site-packages/keras/models.py in add(self, layer)
    330                  output_shapes=[self.outputs[0]._keras_shape])
    331         else:
--> 332             output_tensor = layer(self.outputs[0])
    333             if isinstance(output_tensor, list):
    334                 raise TypeError('All layers in a Sequential model '

/home/pranshu_44/anaconda3/lib/python3.5/site-packages/keras/engine/topology.py in __call__(self, x, mask)
    527             # Raise exceptions in case the input is not compatible
    528             # with the input_spec specified in the layer constructor.
--> 529             self.assert_input_compatibility(x)
    530 
    531             # Collect input shapes to build layer.

/home/pranshu_44/anaconda3/lib/python3.5/site-packages/keras/engine/topology.py in assert_input_compatibility(self, input)
    467                                          self.name + ': expected ndim=' +
    468                                          str(spec.ndim) + ', found ndim=' +
--> 469                                          str(K.ndim(x)))
    470             if spec.dtype is not None:
    471                 if K.dtype(x) != spec.dtype:

ValueError: Input 0 is incompatible with layer convolution2d_11: expected ndim=4, found ndim=2

我正在尝试对CIFAR 10进行图像分类,但是出现了这个错误。根据文档 [https://keras.io/layers/convolutional/][1] ,我的答案是正确的,但我不知道为什么会出现这个错误。

  • 我应该使用(None, 32, 32, 3)的维度吗?

你如何使用你的脚本? - Marcin Możejko
1个回答

1
Conv2D层需要4维的输入,但是显然你只提供了2维。不过我相信你已经注意到了这一点。
根据adventuresinmachinelearning
供给的数据格式应该为[i,j,k,l],其中i是训练样本数,j是图像的高度,k是宽度,l是通道数。
我不熟悉你正在使用的数据,但l(通道数)的值应该为:
对于灰度图像,l始终等于1(如果我们有RGB图像,则等于3)
所以基本上你只需要:
import tensorflow as tf
tf.reshape(your_image_tensor, [-1, 28, 28, 1]) #For a grayscale image
tf.reshape(your_image_tensor, [-1, 28, 28, 3]) #For a RGB image

对自己的代码进行适当的更改。如果您不想使用tensorflow,我建议您阅读this

更新: 您也可以使用numpy.reshape()来重塑数组 了解更多:https://docs.scipy.org/doc/numpy/reference/generated/numpy.reshape.html


网页内容由stack overflow 提供, 点击上面的
可以查看英文原文,
原文链接