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

Question

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

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

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

Answer 1

如果您不介意加载额外的数据，最简单的方法是找出女巫是水果标签，然后执行如下操作:X_train, y_train = X_train[y_train == Fruit_label], y_train[y_train == Fruit_label]，前提是你的数据存储在np.arrays中。相当于您的测试集。

如果没有，那么您将必须修改您的 hdf5 文件或存储数据的位置。