CIFAR-10数据集可视化Python版本

CIFAR-10数据集有3个版本，分别是Python, Matlab, 二进制版本。

切记这里下载Python版本。

编辑代码export_cifar10.py

import pickle
from scipy.misc import imsave
import numpy as np

def unpickle(file):
    f = open(file, 'rb')
    dict = pickle.load(f, encoding='iso-8859-1')
    f.close()
    return dict

filename = '/tmp/cifar10_data/cifar-10-batches-py/test_batch'
dict_train_batch1 = unpickle(filename)

print(dict_train_batch1.keys())  # 字典里有4组键值对

images = dict_train_batch1['data']
print(images.shape)

img = np.reshape(images[10], (3, 32, 32)) #只是测试，没有采用循环导出全部图像，这里导出第10幅图
img = img.transpose(1, 2, 0)

imsave("cifar.jpg", img)