Keras cifar-10 Value error different tensorshape

Question

我在这里学习标准的 cifar 10 keras 教程：https://github.com/fchollet/keras/blob/master/examples/cifar10_cnn.py

我对其进行了修改以使用我自己的训练图像。每张图片都复制了 cifar 集的尺寸，即它们都是 32x32 和 3 个通道。

每张图片的形状：

(32,32,3)

但是，我遇到了 ValueError，如下面的完整输出所示。

X_train shape: (7200, 32, 32, 3)
7200 train samples
800 test samples
Using real-time data augmentation.
Epoch 1/200
Traceback (most recent call last):

  File "<ipython-input-16-70ca8831e139>", line 162, in <module>
    validation_data=(X_test, Y_test))

  File "/storage/programfiles/anaconda3/lib/python3.5/site-packages/keras/models.py", line 651, in fit_generator
    max_q_size=max_q_size)

  File "/storage/programfiles/anaconda3/lib/python3.5/site-packages/keras/engine/training.py", line 1383, in fit_generator
    class_weight=class_weight)

  File "/storage/programfiles/anaconda3/lib/python3.5/site-packages/keras/engine/training.py", line 1167, in train_on_batch
    outputs = self.train_function(ins)

  File "/storage/programfiles/anaconda3/lib/python3.5/site-packages/keras/backend/tensorflow_backend.py", line 659, in __call__
    updated = session.run(self.outputs + self.updates, feed_dict=feed_dict)

  File "/storage/programfiles/anaconda3/lib/python3.5/site-packages/tensorflow/python/client/session.py", line 372, in run
    run_metadata_ptr)

  File "/storage/programfiles/anaconda3/lib/python3.5/site-packages/tensorflow/python/client/session.py", line 625, in _run
    % (np_val.shape, subfeed_t.name, str(subfeed_t.get_shape())))

ValueError: Cannot feed value of shape (32, 32, 32, 3) for Tensor 'convolution2d_input_7:0', which has shape '(?, 3, 32, 32)'

谁能帮帮我？ :)

编辑： 我尝试如下重塑：

X_train = X_train.reshape((7200,3,32,32))
X_test = X_test.reshape((-1,3,32,32))

它反而崩溃了。

Answer 1

您实际上需要将数组转换为正确的顺序，而不是重塑：

X_train = np.transpose(X_train, (0, 3, 1, 2))
X_test = np.transpose(X_test, (0, 3, 1, 2))