读取cifar10数据集的代码

读取数据集的代码

from six.moves import cPickle as pickle
import os
import platform
import numpy as np
import pickle


filename =r'D:\PythonProjects\cifar-10-batches-py\data_batch_1'  # cifar10二进制文件路径
classes = ('plane', 'car', 'bird', 'cat',
           'deer', 'dog', 'frog', 'horse', 'ship', 'truck')


def load_pickle(f):
    version = platform.python_version_tuple()
    if version[0] == '2':
        return pickle.load(f)
    elif version[0] == '3':
        return pickle.load(f, encoding='latin1')
    raise ValueError("invalid python version: {}".format(version))


def load_CIFAR_batch(filename):
    """ load single batch of cifar """
    with open(filename, 'rb') as f:
        datadict = load_pickle(f)
        X = datadict['data']
        Y = datadic