如何在keras中使用ImageDataGenerator和flow_from_directory保存增强的图像

from keras.preprocessing.image import ImageDataGenerator
from keras.preprocessing.image import array_to_img, img_to_array, load_img

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'     # 填充像素模式
)

img_dir = 'path/to/images_directory'
save_dir = 'path/to/save_directory'
batch_size = 32

img_generator = datagen.flow_from_directory(
    img_dir,                  # 图像目录路径
    target_size=(224, 224),   # 图像尺寸
    batch_size=batch_size,    # 批处理大小
    class_mode='categorical', # 分类模式
    save_to_dir=save_dir,     # 保存增强后图像的路径
    save_prefix='augmented',  # 保存图像的前缀
    save_format='jpg'         # 保存图像的格式
)

# 保存增强后的图像
for i in range(len(img_generator)):
    img, _ = img_generator[i]
    for j in range(batch_size):
        save_path = os.path.join(save_dir, f"augmented_{i*batch_size + j}.jpg")
        img_to_array(img[j]).save(save_path)