这个保存就很简单,因为使用过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资源下载