Keras：使用 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)

如何绘制带有数据增强示例的示例图像？

比如我的目录是../input/train/cats/和../input/train/dogs/

Answer 1

生成器将只为您提供增强后的 imgs，而不是没有

如果您想查看图像之间的差异，我会：

1. 按字母数字顺序排列名称为 00001.jpg,00002.jpg .....
2. 使用 shuffle=False，batch_size=1

所以生成器将按照给定的顺序生成 imgs。然后生成器会给你的第一个 img 是 00001.jpg 的增强，第二个到 00002.jpg 等等。