получить xtrain, y train, xtest и ytrain из моих собственных данных изображений

Question

Мои данные упорядочены в папках следующим образом:

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

Содержит 1750 64x64 img на папку в поезде и 250 p / папок в тесте. Я хочу получить переменные xtest / ytest / xtrain / ytrain из этих данных. Но я могу найти способ сделать это, например, из CIFAR-10:

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

Есть мысли? Спасибо.