Keras CIFAR-10值误差不同tensorshape

我是按照标准CIFAR 10 keras教程这里：https://github.com/fchollet/keras/blob/master/examples/cifar10_cnn.py Keras CIFAR-10值误差不同tensorshape

其中我修改，用我自己的训练图像。每个图像复制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))

坠毁代替。

来源

2016-07-06 Wboy

实际上，你需要你的阵列调换到正确的排序，而不是一个重塑：

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

来源

2016-07-12 05:33:05

Keras CIFAR-10值误差不同tensorshape

回答

相关问题