从这里可以看到如何使用现有的图像集cifar10进行图像增强。

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

回答：

你可以将你的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文档在这里