如果您打印,x_train.shape
您将看到形状正在(50000, 32, 32, 3)
您input_shape=(3, 32, 32)
在第一层中给出的形状.该错误只是说预期的输入形状和给定的数据是不同的.
你需要做的就是给予 input_shape=(32, 32, 3)
.此外,如果您使用此形状,则必须使用tf
您的图像排序.backend.set_image_dim_ordering('tf')
.
否则,您可以置换数据轴.
x_train = x_train.transpose(0,3,1,2) x_test = x_test.transpose(0,3,1,2) print x_train.shape