python-3.x - 如何解开 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()

文档在这里。它应该可以帮助您解决所有 3 个问题，因为 keras 会为您处理一切。