keras实现MNIST数据集的第4个图像显示——ValueError: cannot reshape array of size 47040000 into shape (6000,28,28)

最新推荐文章于 2023-09-04 10:47:49 发布

我是小杨我就这样

最新推荐文章于 2023-09-04 10:47:49 发布

阅读量2.2k

点赞数

分类专栏：深度学习框架—Keras

本文链接：https://blog.csdn.net/weixin_44478378/article/details/102473808

版权

深度学习框架—Keras 专栏收录该内容

18 篇文章 2 订阅

订阅专栏

"""注意：
       mnist数据集中图像的存储形状为28*28的灰度图像即一维
       plt.imshow显示为二维图像，故需要将一维数组更改为二维数组
       而且要针对对应的那个图像进行形状更改
       """
print(train_images[4].shape)
train_images_4 = train_images[4].reshape(28, 28)
print(train_images_4.shape)

plt.imshow(train_images_4, cmap=plt.cm.binary)
plt.show()