CIFAR-10数据集可视化Python版本

最新推荐文章于 2025-06-12 21:43:30 发布

黑虎含珠

最新推荐文章于 2025-06-12 21:43:30 发布

阅读量2.5k

点赞数

CC 4.0 BY-SA版权

分类专栏： # TensorFlow

本文链接：https://blog.youkuaiyun.com/chenhaifeng2016/article/details/72582133

TensorFlow 专栏收录该内容

43 篇文章

订阅专栏

本文介绍如何使用Python加载CIFAR-10数据集，并展示了从数据集中读取图像的具体步骤。通过提供的Python脚本，可以实现对数据集的解析并显示其中的一张图片。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

CIFAR-10数据集有3个版本，分别是Python, Matlab, 二进制版本。

切记这里下载Python版本。

编辑代码export_cifar10.py

import pickle
from scipy.misc import imsave
import numpy as np

def unpickle(file):
    f = open(file, 'rb')
    dict = pickle.load(f, encoding='iso-8859-1')
    f.close()
    return dict

filename = '/tmp/cifar10_data/cifar-10-batches-py/test_batch'
dict_train_batch1 = unpickle(filename)

print(dict_train_batch1.keys())  # 字典里有4组键值对

images = dict_train_batch1['data']
print(images.shape)

img = np.reshape(images[10], (3, 32, 32)) #只是测试，没有采用循环导出全部图像，这里导出第10幅图
img = img.transpose(1, 2, 0)

imsave("cifar.jpg", img)