如何根据特定的类名加载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文件或任何你存储数据的地方。