官方网站下载的文件 cifar-10-python 解压得到的数据如下:
可见并不是原始的图片数据;由于训练需要,这里需要转成图片格式。参考代码:https://www.cnblogs.com/dzzy/p/10824072.html
唯一问题是修改了该文章使用的 import scipy.misc,存在一点问题因此做了修改:引入imageio包
# import scipy.misc
import numpy as np
import imageio # 引入imageio包
# 解压 返回解压后的字典
def unpickle(file):
import pickle as pk
fo = open(file, 'rb')
dict = pk.load(fo, encoding='iso-8859-1')
fo.close()
return dict
# 生成训练集图片
for j in range(1, 6):
dataName = "./data/imagenet/train/train/data_batch_" + str(j) # 读取当前目录下的da