如何根据特定的类名加载 CIFAR-10 数据集？答案

【问题标题】：How to load CIFAR-10 datasets based on specific class name?如何根据特定的类名加载 CIFAR-10 数据集？
【发布时间】：2016-11-18 01:29:24
【问题描述】：

我正在使用 CIFAR-10 数据集进行深度学习，但我只想为水果类指定我的数据集。我们知道我们使用过：

(X_train, y_train), (X_test, y_test) = cifar10.load_data()

加载所有 CIFAR-10 数据集。如何只加载水果类的数据而不是全部数据？

【问题讨论】：

加载数据时，模型没有经过训练，无法分辨哪些图片有成果。
是否可以根据 CIFAR 的类别标签指定图像？
你只需要知道水果类的标签。然后根据y_train和y_test中的标签过滤数据

标签： python machine-learning tensorflow deep-learning keras

【解决方案1】：

如果您不介意加载额外的数据，最简单的方法是找出 wwitch 是水果标签并执行以下操作：X_train, y_train = X_train[y_train == fruit_label], y_train[y_train == fruit_label]，前提是您的数据存储在 np.arrays 中。等效于您的测试集。

如果没有，那么您将不得不修改您的 hdf5 文件或您存储数据的任何位置。

【讨论】：