初识Cifar10

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

import pickle
import numpy as np
import cv2
import os

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

label_name=["airplane","automobile","bird","cat","deer","dog","frog","horse","ship","truck"]

import glob

train_list=glob.glob("cifar-10-python/test_batch")
print(train_list)
save_path='cifar-10-python/test'

for l in train_list:
    print(l)
    l_dict=unpicle(l)
    print(l_dict.keys())
    for im_idx,im_data in enumerate(l_dict[b'data']):
        im_label=l_dict[b'labels'][im_idx]
        im_name=l_dict[b'filenames'][im_idx]
        #print(im_label,im_name)
        im_label_name=label_name[im_label]
        im_data=np.reshape(im_data,[3,32,32])
        im_data=np.transpose(im_data,(1,2,0))
        # cv2.imshow("im_data",cv2.resize(im_data,(100,100)))
        # cv2.waitKey(0)

        if not os.path.exists("{}/{}".format(save_path,im_label_name)):
            # 如果文件夹不存在 则创建
            os.mkdir("{}/{}".format(save_path,im_label_name))
            print(im_label_name+"directory make successfully")

        cv2.imwrite("{}/{}/{}".format(save_path,im_label_name,im_name.decode("utf-8")),im_data)