從我自己的圖像數據中獲得xtrain，y train，xtest和ytrain

Question

我的數據在文件夾中的排序如下：

- train
    * principal
    * soup
    * dessert
- test
    * principal
    * soup
    * dessert

它在火車中每個文件夾包含1750 64x64 img，在測試中包含250 p /文件夾。 我想從該數據獲取xtest / ytest / xtrain / ytrain vars。 但是我只能從CIFAR-10中找到一種方法來做到這一點：

(x_train, y_train), (x_test, y_test) = cifar10.load_data() 
# x_train - training data(images), y_train - labels(digits)

任何想法？ 謝謝。