如何将MNIST数据集数据图像可视化

Posted 2022-09-27 算法与编程之美

1 问题

怎么将MNIST数据集里的数字图像数据可视化？

2 方法

基于pytorch框架，torchvision.datasets.MNIST导入数据集，对图片数据进行预处理以及使用matplotlib可视化。

步骤:

1. import 文件
   from torchvision import datasets
   import matplotlib.pyplot as plt
   import os
   import numpy as np
2. 导入数据
   #训练集
   train_ds =datasets.MNIST(
       root='data',#说明数据集下载的路径
       download=True,
       train=True
   )
   #测试集
   test_ds=datasets.MNIST(
       root='data',
       download=True,
       train=False
3. 可视化

ds_images,ds_targets=(test_ds._load_data())

fig = plt.figure()#生成图框

for i,c in enumerate(np.random.randint(0,1000,25)):#随机取0，1000里的25张图片

    plt.subplot(5,5,i+1)

    plt.tight_layout()#调整间距

    plt.imshow(ds_images[c], interpolation='none')

    plt.title("数字标签: ".format(ds_targets[c]))

    plt.rcParams['font.sans-serif']=['SimHei']

plt.show()

_load_data 返回 data 和 target 正是我们所需得到的图像和标记、

可视化结果：

3 结语

针对该问题使用了基于pytorch框架，torchvision.datasets.MNIST导入数据集，用test_ds._load_data来获取图字的标记数字，用matplotlib来进行图像可视化显示。