mnist数据集读取与显示的代码
时间: 2023-11-27 17:22:49 浏览: 90
以下是使用Python和OpenCV库读取MNIST数据集并显示图像的示例代码:
```python
import cv2
import numpy as np
import gzip
# 读取MNIST数据集的图像和标签
def read_mnist_data(images_path, labels_path):
with gzip.open(labels_path, 'rb') as labels_file:
labels = np.frombuffer(labels_file.read(), dtype=np.uint8, offset=8)
with gzip.open(images_path, 'rb') as images_file:
images = np.frombuffer(images_file.read(), dtype=np.uint8, offset=16).reshape(len(labels), 28, 28)
return images, labels
# 显示MNIST数据集中的图像
def show_mnist_image(image):
cv2.imshow('MNIST Image', image)
cv2.waitKey(0)
cv2.destroyAllWindows()
# 读取MNIST数据集的图像和标签
images_path = 'train-images-idx3-ubyte.gz'
labels_path = 'train-labels-idx1-ubyte.gz'
images, labels = read_mnist_data(images_path, labels_path)
# 显示第一张图像
show_mnist_image(images[0])
```
请注意,以上代码仅演示了如何读取MNIST数据集的图像和标签以及如何显示图像。在实际应用中,您需要将数据集拆分为训练集和测试集,并将其用于训练和测试机器学习模型。
阅读全文