python把图像批量数据增强的代码

以下是一个使用Python代码对图像进行数据增强的示例：

from keras.preprocessing.image import ImageDataGenerator
import os

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')

# 设置原始图像的路径和目标路径
image_path = 'path/to/original/image'
save_path = 'path/to/saved/image'

# 获取所有图像的文件名
files = os.listdir(image_path)

# 迭代所有图像并进行数据增强
for file in files:
    img = load_image(image_path + file)
    img = img.reshape((1,) + img.shape)
    i = 0
    for batch in datagen.flow(img, batch_size=1,
                              save_to_dir=save_path, save_prefix=file.split('.')[0], save_format='jpg'):
        i += 1
        if i > 5:
            break

这段代码使用了Keras中的ImageDataGenerator类来实现图像增强，包括旋转、位移、剪切、缩放、水平翻转等操作。