cifar10数据集下载及图片格式解析

CIFAR-10 是由 Hinton 的学生 Alex Krizhevsky 和 Ilya Sutskever 整理的一个用于识别普适物体的小型数据集。一共包含 10 个类别的 RGB 彩色图片：飞机（ a叩lane ）、汽车（ automobile ）、鸟类（ bird ）、猫（ cat ）、鹿（ deer ）、狗（ dog ）、蛙类（ frog ）、马（ horse ）、船（ ship ）和卡车（ truck ）。

图片的尺寸为 32×32×3 ，数据集中一共有 50000 张训练圄片和 10000 张测试图片。CIFAR-10 的图片样例如图所示。

cifar10数据集是一个常用的小型物体数据集，很多模型拿该数据集进行跑流程的简单测试。下面是通过pytorch官方代码自动下载cifar10数据集，并且python解析保存为img格式。

注意解析为img时候，root_dir中原始cifar-10-batches-py文件路径-不识别，需要修改一下。

保存的img路径提前建好文件夹。


import torchvision
from torch.utils.data import DataLoaderimport numpy as np
import imageio  # 引入imageio包train_data = torchvision.datasets.CIFAR10(root="dataset", train=True, transform=torchvision.transforms.ToTensor(), download=True)
test_data = torchvision.datasets.CIFAR10(root="dataset", train=False, transform=torchvision.transforms.ToTensor(), download=True)# 解压 返回解压后的字典
def unpickle(file):import pickle as pkfo = open(file, 'rb')dict = pk.load(fo, encoding='iso-8859-1')fo.close()return dict#begin unpickle
root_dir="./dataset/"
# 生成训练集图片
for j in range(1, 6):dataName = root_dir+"/data_batch_" + str(j)  # 读取当前目录下的data_batch1~5文件。Xtr = unpickle(dataName)print(dataName + " is loading...")for i in range(0, 10000):img = np.reshape(Xtr['data'][i], (3, 32, 32))  # Xtr['data']为图片二进制数据img = img.transpose(1, 2, 0)  # 读取imagepicName = root_dir+'/train/' + str(Xtr['labels'][i]) + '_' + str(i + (j - 1) * 10000) + '.jpg'imageio.imsave(picName, img)  # 使用的imageio的imsave类print(dataName + " loaded.")print("test_batch is loading...")# 生成测试集图片
testXtr = unpickle(root_dir+"/test_batch")
for i in range(0, 10000):img = np.reshape(testXtr['data'][i], (3, 32, 32))img = img.transpose(1, 2, 0)picName = root_dir+'/test/' + str(testXtr['labels'][i]) + '_' + str(i) + '.jpg'imageio.imsave(picName, img)
print("test_batch loaded.")

cifar10数据集下载及图片格式解析相关推荐

CIFAR10数据集转化为图片
CIFAR10下载:http://www.cs.toronto.edu/~kriz/cifar.html 1. 数据集介绍该数据集共有60000张彩色图像,这些图像是32*32,分为10个类,每类6 ...
MNIST数据集转为.jpg图片格式
从mnist官网下载下来的mnist手写数据集是二进制文件流格式的,不能直接查看,如果需要查看,需要将二进制文件转化为jpg格式,可以用各种编程语言实现,如MATLAB.Python.C++等,本文是 ...
python 将MNIST数据集转为jpg图片格式
下载的数据集格式是字节存储的,有时需要转为图片格式,以下以测试集为例子,说明python转换代码. IDX数据格式这四个文件采用了IDX的文件格式,一种平铺直叙的方式: magic number s ...
常用图标png、ico 图标下载，图片格式转换为ico
网上找到几个不错的免费下载地址: 1.http://www.easyicon.net/ 2.http://588ku.com/sucai/ 3.http://www.iconpng.com 可以直接下 ...
cifar-100数据集转成图片格式
cifar-100数据集转成png格式 import os import numpy as np from scipy import misc# 给我个路径我要创建文件呢 def makfile(pa ...
CIFAR-10数据集下载
文章目录下载使用说明 Python / Matlab versions Binary version 下载 Cifar10是一个小型的图片分类的数据集,不详细介绍,参照官网: 这里是官网 www ...
CIFAR10数据集
目录一.CIFAR10数据集简介: 二.CIRAR10数据集格式三.CIFAR10数据集下载与读取 1.下载: 2.读取四.class:torchvision.datasets.CIFAR10 ...
【Python2】Keras_ResNet 在Cifar10数据集上分类，Flask框架部署目标检测模型
文章目录 1.导入库 2.数据准备 2.1 加载训练集 2.2 加载测试集 2.3 对类别做One-Hot编码 2.4 对图片像素的0-255值做归一化,并减去均值 3.搭建神经网络 3.1 定义函数 ...
cifar10数据集训练
有关CIFAR-10数据集 (1)CIFAR-10数据集由10个类的60000个32x32彩色图像组成,每个类有6000个图像.有50000个训练图像和10000个测试图像. (2)数据集分为五个训 ...

cifar10数据集下载及图片格式解析

cifar10数据集下载及图片格式解析相关推荐

最新文章

热门文章