如何解开CIFAR-10，加载批次和拆分数据集？

Question

使用卷积神经网络进行CIFAR-10数据集分类。我无法理解：

1）在何处提取CIFAR-10数据集文件 2）如何使用pickle框架加载批次 3）将数据集分为训练数据和测试数据

请帮助代码，在jupyter笔记本上使用python 3.6。

我尝试过此方法，但似乎无济于事。可能是因为我将CIFAR-10文件保存在错误的位置。

def unpickle(file):
    import pickle
    with open(file, 'rb') as fo:
        dict = pickle.load(fo, encoding='bytes')
    return dict

Answer 1

您可以尝试使用keras数据集加载数据：

from keras.datasets import cifar10

(x_train, y_train), (x_test, y_test) = cifar10.load_data()

文档为here。 它可以帮助您解决所有3个问题，因为keras可以为您解决所有问题。