python数据增强代码

以下是一个简单的 Python 数据增强代码示例：

import cv2
import numpy as np
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'  # 填充模式
)

# 加载图像
img = cv2.imread('image.jpg')

# 将图像转换为数组
x = np.expand_dims(img, axis=0)

# 生成增强后的图像
i = 0
for batch in datagen.flow(x, batch_size=1):
    cv2.imwrite('image_augmented_%d.jpg' % i, batch[0])
    i += 1
    if i == 5:  # 生成 5 张增强后的图像
        break

这段代码使用 Keras 的 ImageDataGenerator 类定义了一个数据增强器，然后使用 OpenCV 加载图像，并将其转换为数组。最后，使用数据增强器生成 5 张增强后的图像，并保存到文件中。