如何查看mnist数据集的图片

📚博客主页：knighthood2001

✨公众号：认知up吧 （目前正在带领大家一起提升认知，感兴趣可以来围观一下）

🎃知识星球：【认知up吧|成长|副业】介绍

❤️感谢大家点赞👍🏻收藏⭐评论✍🏻，您的三连就是我持续更新的动力❤️

🙏笔者水平有限，欢迎各位大佬指点，相互学习进步！

今天来讲讲如何查看mnist数据集的图片 

import numpy as np
import matplotlib.pyplot as plt


def read_mnist_images(filename):
    with open(filename, 'rb') as f:
        # 读取魔术数字、图像数量、行数、列数
        magic_number = int.from_bytes(f.read(4), 'big')
        number_of_images = int.from_bytes(f.read(4), 'big')
        rows = int.from_bytes(f.read(4), 'big')
        cols = int.from_bytes(f.read(4), 'big')

        # 读取图像数据
        images = np.frombuffer(f.read(), dtype=np.uint8)
        images = images.reshape((number_of_images, rows, cols))
    return images


# 修改为您的路径
filename = './data/MNIST/raw/train-images-idx3-ubyte'

images = read_mnist_images(filename)

# 显示第一张图像
plt.imshow(images[0], cmap='gray')
plt.title("First Image in MNIST Dataset")
plt.show()

把最后第三行的0，修改成别的数字，就会显示某张数字。 

运行之后，结果如下