如何使用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