从torchvision加载FashionMNIST数据并用matplotlib画出来

环境:

Python版本： 3.8
IDE：Spyder 5.2.2

代码：

from torchvision import datasets
from matplotlib import pyplot as plt

# 下载数据
training_data = datasets.FashionMNIST(
    root='data',  #这个root指的是数据存放在本地电脑的路径
    train=True,   #拿的是训练的数据
    download=True  #是否要下载
)

test_data = datasets.FashionMNIST(
    root='data',
    train=False,  
    download=True
    )

tdata = test_data.data.numpy() # tensor转换成numpy

for i in range(len(tdata)):
    # 每隔100张画一张
    if i % 100 == 99:
        print(i)
        plt.figure(i)
        plt.imshow(tdata[i])
    

效果