1. 程式人生 > >cifar10 資料集轉換成圖片

cifar10 資料集轉換成圖片

自己親測,轉換一部分,可以使用

import cv2
import numpy as np
file = 'data_batch_2'

官方給的示例:

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

得到資料字典

dict1 = unpickle(file)
for i in range(100):#我只儲存的100張
    img = dict1["data"][i]#得到圖片的資料
    img = np.reshape(img, (3, 32,32))  #轉為三維圖片陣列
    img = img.transpose((1,2,0))#通道轉換為CV2的要求形式
    img_name = dict1["filenames"][i]#拿到圖片的名字
    img_label = str(dict1["labels"][i])#拿到圖片的標籤
    cv2.imwrite("/home/user/3/"+img_label+"_"+img_name,img)#儲存