一、ImageDataGenerator
def image_preprocess():
# 训练集的图片生成器,通过参数的设置进行数据扩增
train_datagen = ImageDataGenerator(
preprocessing_function=preprocess_input,
rotation_range=30,
width_shift_range=0.2,
height_shift_range=0.2,
shear_range=0.2,
zoom_range=0.2,
horizontal_flip=True
)
# 验证集的图片生成器,不进行数据扩增,只进行数据预处理
val_datagen = ImageDataGenerator(
preprocessing_function=preproce