ImageDataGenerator ()用法简介
ImageDataGenerator
是 Keras 中用于图像数据增强和预处理的工具。它允许你实时生成增强的图像数据,这在训练深度学习模型时特别有用,因为它可以帮助提高模型的泛化能力。
以下是 ImageDataGenerator
的一些常见用法和参数:
基本用法
-
导入模块
from keras.preprocessing.image import ImageDataGenerator
-
创建 ImageDataGenerator 实例
你可以通过指定各种参数来配置数据增强方式。例如:
datagen = ImageDataGenerator(rotation_range=40, # 随机旋转的度数范围width_shift_range=0.2, # 水平平移的范围(相对于总宽度的比例)height_shift_range=0.2, # 垂直平移的范围(相对于总高度的比例)shear_range=0.2, # 随机错切变换的角度zoom_range=0.2, # 随机缩放的范围horizontal_flip=True, # 水平翻转fill_mode='nearest' # 填充新创建像素的方法 )
-
使用
flow
或flow_from_directory<