从我自己的图像数据中获得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)

任何想法？ 谢谢。