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