mnist torch加载fashion_FashionMNIST数据集保存为图片

这个保存就很简单，因为使用过pytorch里面加载FashionMNIST数据集，就会生成这样一个文件

里面有两个文件

于是加载就非常简单了

import torch

import cv2

import numpy as np

train_data=torch.load("training.pt")[0].numpy()

test_data=torch.load("test.pt")[0].numpy()

print(train_data.shape)

print(test_data.shape)

#print(test_data[0])

for i in range(train_data.shape[0]):

cv2.imwrite("train\\"+str(i)+".jpg",train_data[i][:,:,np.newaxis])

for i in range(test_data.shape[0]):

cv2.imwrite("test\\"+str(i)+".jpg",test_data[i][:,:,np.newaxis])

运行之前自己在路径下新建train和test文件夹

运行结果

友情提示：如想要处理好的图片文件和源文件，可以去我的csdn资源下载