如何使用自定义数据生成器进行keras图像增强？

from keras.preprocessing.image import ImageDataGenerator

datagen = ImageDataGenerator(
    rotation_range=20,  # 随机旋转角度范围
    width_shift_range=0.2,  # 随机水平平移范围
    height_shift_range=0.2,  # 随机垂直平移范围
    shear_range=0.2,  # 随机错切变换范围
    zoom_range=0.2,  # 随机缩放范围
    horizontal_flip=True,  # 随机水平翻转
    fill_mode='nearest'  # 填充像素的策略
)

train_generator = datagen.flow_from_directory(
    'path/to/train_directory',  # 训练集目录
    target_size=(224, 224),  # 图像尺寸
    batch_size=32,  # 批量大小
    class_mode='binary'  # 分类模式
)

model.fit_generator(
    train_generator,  # 训练数据生成器
    steps_per_epoch=2000,  # 每个epoch的步数
    epochs=50  # 训练的总epoch数
)