如何根據特定的類名加載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文件或無論你在哪裏存儲你的數據。