cafir10分类图片转化为png图片（以文件夹分类好）

最新推荐文章于 2022-06-24 10:54:36 发布

华天雪 L.W

最新推荐文章于 2022-06-24 10:54:36 发布

阅读量971

点赞数 1

分类专栏： python

本文链接：https://blog.csdn.net/huatianxue/article/details/109457331

版权

python 专栏收录该内容

97 篇文章 6 订阅

订阅专栏

import cv2
import numpy as np
file = './data/cifar-10-batches-py/test_batch'
# print(file)
def unpickle(file):
    # import _pickle as cPickle
    import pickle
    with open(file, 'rb') as fo:
        dict = pickle.load(fo,encoding="latin1")
    return dict


dict1 = unpickle(file)

for i in range(10000):
    img = dict1["data"][i]
    img = np.reshape(img, (3, 32, 32))
    img = img.transpose((1, 2, 0))
    img_name = dict1["filenames"][i]
    img_label = str(dict1["labels"][i])
    my_path = "./data/png/test/" + img_label + "/" + img_name
    # 这个是关键，相当于存到对应文件夹下了
    cv2.imwrite(my_path, img)