cifar-10数据集的下载及恢复原图片

最新推荐文章于 2025-09-06 19:16:29 发布

原创

最新推荐文章于 2025-09-06 19:16:29 发布 · 4.2k 阅读

16 ·

CC 4.0 BY-SA版权

文章标签：

#深度学习 #人工智能 #计算机视觉

这篇博客详细介绍了如何下载CIFAR-10数据集，解压并将其转化为图片，包括训练集和测试集的处理过程。通过Python实现，展示了如何使用pickle和imageio库操作数据，最终生成按类别组织的训练和验证图片文件夹。

数据集下载

进入官网CIFAR-10 and CIFAR-100 datasets

选择python版本

解压后得到文件夹：

data_batch_1~data_batch_5是训练集数据，每个文件里有10000张图片，test_batch是测试集数据，也含有10000张图片。

转成图片

import numpy as np
import pickle
import imageio
import os


# 解压缩，返回解压后的字典
def unpickle(file):
    fo = open(file, 'rb')
    dict = pickle.load(fo, encoding='latin1')
    fo.close()
    return dict


list = ["airplane", "automobile", "bird", "cat", "deer", "dog", "frog", "horse", "ship", "truck"]
for z in range(10):
    i