cifar10 数据集介绍

最新推荐文章于 2022-11-30 18:19:22 发布

原创

最新推荐文章于 2022-11-30 18:19:22 发布 · 8.4k 阅读

12 ·

CC 4.0 BY-SA版权

CIFAR-10数据集包含60000张32*32彩色图片，分为10个类别，如飞机、汽车等。它分为5万张训练图和1万张测试图。可以从 http://www.cs.toronto.edu/~kriz/cifar.html 下载。

基本信息

CIFAR-10 是一个包含60000张图片的数据集。其中每张照片为32*32的彩色照片，每个像素点包括RGB三个数值，数值范围 0 ~ 255。所有照片分属10个不同的类别，分别是 'airplane', 'automobile', 'bird', 'cat', 'deer', 'dog', 'frog', 'horse', 'ship', 'truck'。其中五万张图片被划分为训练集，剩下的一万张图片属于测试集。

下载数据集

打开下面的链接进入官网下载

http://www.cs.toronto.edu/~kriz/cifar.html

查看文件

基于Python3.5

#python3
import numpy as np
import pickle
import os

CIFAR_DIR = "./cifar-10-batches-py"
print(os.listdir(CIFAR_DIR))

with open(os.path.join(CIFAR_DIR, "data_batch_1"), 'rb') as f:
	data = pickle.load(f, encoding='bytes')
	print(type(data))
	print(data.keys())
	print(type(data[b'data']))
	print(type(data[b'labels']))
	print(type(data[b'batch