如果我通过以下方式在Keras中下载CIFAR 10图像:
现在我通过Keras获取MNIST图像的方法如下:
(x_train, y_train), (x_test, y_test) = cifar10.load_data()
# Getting shape
x_train.shape
>>> (50000, 32, 32, 3)
我可以通过执行以下步骤绘制每个图像:
# Plot RGB image
plt.imshow(x_train[0])
# Plot only one colour-channel e.g. R
plt.imshow(x_train[0][:,:,0])
现在我通过Keras获取MNIST图像的方法如下:
(X_train, y_train), (X_test, y_test) = mnist.load_data()
# getting shape
X_train.shape
>>> (60000, 28, 28)
然而,由于它是灰度图像,深度通道应该为1。如果我使用以下方法重塑它:
X_train = X_train.reshape(X_train.shape[0],28,28,1)
我可以让神经网络工作,但无法再绘制它。应该如何正确地调整其形状,以便仍然可以绘制图形?