如何将ImageDataGenerator与我自己的图像集一起用于图像增强

Question

来自here 它显示了如何使用现有图像集cifar10进行图像增强。

如何使用我自己的图像集创建ImageDataGenerator？

Answer 1

您可以将train_generator更改为flow_from_directory()，只需指向您选择的图片目录：

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

您需要按类别将图片放在子文件夹中，即your/data/here/class1，your/data/here/class2等。

Keras docs here