如何从keras加载我的数据集作为示例？

Question

我使用Keras example中的代码，在那里，数据作为行加载

(x_train, y_train), (x_test, y_test) = cifar10.load_data()

如何将我的图像加载到此示例中？我试图找到实现，但是所有内容都是基于开放数据集的，其加载方式与Keras示例相同。

Answer 1

您可以创建带有训练图像的文件夹保存，并使用flow_from_directory（）：

train_generator = train_datagen.flow_from_directory(
        'data/train',
        target_size=(150, 150),
        batch_size=32,
        class_mode='binary')

您可以在此处找到更多信息：https://keras.io/preprocessing/image/