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

最新推荐文章于 2024-11-07 23:54:38 发布

原创最新推荐文章于 2024-11-07 23:54:38 发布 · 1k 阅读

2 ·

本内容遵循CC 4.0 BY-SA版权协议

收录于

python 专栏收录该内容

97 篇文章

订阅专栏

博客主要围绕将CIFAR10分类图片转化为PNG图片，并按文件夹进行分类的内容展开，涉及信息技术中图片处理与分类相关操作。

开发板推荐：天空星STM32F407VET6开发板

超高性价比 STM32主控 | 超高主频 | 一板兼容百芯 | 比赛神器 | 沉金彩色丝印

点击查看

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)