【发布时间】:2018-11-17 04:00:59
【问题描述】:
我正在 Keras 中使用 ImageDataGenerator 和 flow_from_directory 训练深度神经网络。数据在一个文件夹中。因此,我在使用ImageDataGenerator 创建生成器时使用validation_split=0.x。然后我创建了两个流,一个用于训练,一个用于验证,分别使用 flow_from_directory 和 subset="training" 和 subset="validation"。
我想知道在创建 ImageDataGenerator 时是否将任何指定的图像增强(转换)应用于训练和验证子集或仅应用于训练子集。
我在 GitHub 的 Keras 存储库中找不到正确的部分来检查它。
(注意:我知道最好使用两个单独的目录来使用两个单独的生成器进行训练和验证)
代码示例:
img_gen = ImageDataGenerator(validation_split=0.2,horizontal_flip = True, vertical_flip = True,...)
train_flow = img_gen.flow_from_directory('directory',subset = "training",...)
validation_flow = img_gen.flow_from_directory('directory',subset = "validation",...)
history=model.fit_generator(generator = train_flow ,validation_data = validation_flow,...)
【问题讨论】:
标签: python tensorflow keras deep-learning