Keras：Plot 使用 flow_from_directory 訓練數據的圖像增強樣本

Question

我正在使用 Keras flow_from_directory 來執行數據增強：

from keras.applications.inception_v3 import preprocess_input
from keras.preprocessing.image import ImageDataGenerator

batch_size = 32

# Create train generator
train_datagen = ImageDataGenerator(preprocessing_function=preprocess_input, 
                                  rotation_range = 15,
                                  width_shift_range = 0.2,
                                  height_shift_range = 0.2,
                                  shear_range = 10,
                                  brightness_range = (0.8,0.9)) 

train_generator = train_datagen.flow_from_directory(directory = '../input/train/',
                                                    target_size=(299,299),
                                                    color_mode='rgb',
                                                    batch_size=batch_size,
                                                    class_mode='binary', # if not binary, set to categorical
                                                    shuffle = True, seed = 0)

如何 plot 示例圖像及其數據增強示例？

例如，我的目錄是../input/train/cats/和../input/train/dogs/

Answer 1

生成器只會在增強后為您提供 imgs，而不是沒有

如果您想查看圖像之間的差異，我會：

1. 按字母數字順序排列帶有名稱的img，因此00001.jpg，00002.jpg.....
2. 使用 shuffle=False, batch_size=1

所以生成器將按照給定的順序生成 imgs。 然后生成器會給你的第一個 img 是 00001.jpg 的增強，第二個是 00002.jpg 等等。