cifar10提取目录格式和脚本

官网下载地址下载地址：http://www.cs.toronto.edu/~kriz/cifar-10-python.tar.gz
提取后的下载地址：https://download.csdn.net/download/Night_MFC/12713315

import os
import matplotlib.pyplot as plt

def unpickle(file):
    import pickle
    with open(file, 'rb') as fo:
        dict = pickle.load(fo, encoding='bytes')
    return dict

def save_img(parent, label, filename, img):
    folder = os.path.join(parent, label)
    if not os.path.exists(folder):
        os.makedirs(folder)
    path = os.path.join(parent, label, filename)
    plt.imsave(path, img)


classes = ('plane', 'car', 'bird', 'cat',
           'deer', 'dog', 'frog', 'horse', 'ship', 'truck')

for i in range(1, 6):
    file = 'data_batch_{}'.format(i)
    print(file)
    dict = unpickle(file)
    for i in range(10000):
        img = dict[b'data'][i].reshape((3, 32, 32)).transpose((1, 2, 0))
        label = classes[dict[b'labels'][i]]
        filename = bytes.decode(dict[b'filenames'][i])
        # plt.imshow(img)
        # print(label)
        save_img('train', label, filename, img)
        # break;

dict = unpickle('test_batch')
for i in range(10000):
    img = dict[b'data'][i].reshape((3, 32, 32)).transpose((1, 2, 0))
    label = classes[dict[b'labels'][i]]
    filename = bytes.decode(dict[b'filenames'][i])
#     plt.imshow(img)
#     print(label)
    save_img('test', label, filename, img)