mnist数据集读取与显示代码实现
时间: 2023-11-15 09:25:16 浏览: 167
读取mnist数据集并保存成图片代码
5星 · 资源好评率100%
以下是一个基于Python的MNIST数据集读取与显示的代码示例:
```python
import numpy as np
import matplotlib.pyplot as plt
from tensorflow.keras.datasets import mnist
# 加载MNIST数据集
(x_train, y_train), (x_test, y_test) = mnist.load_data()
# 显示第一张图像
plt.imshow(x_train[0], cmap='gray')
plt.show()
```
代码中使用了Python的`tensorflow.keras.datasets`库中的`mnist`模块来加载MNIST数据集。数据集包含了60000张训练集图像和10000张测试集图像,每张图像大小为28x28像素,且为灰度图像。通过`imshow`函数将第一张图像显示出来,参数`cmap='gray'`表示将图像显示为灰度图像。
阅读全文