帮我用python写一个数据增强jpg格式的代码

当然可以！以下是一个使用Python和OpenCV库进行图像增强的示例代码：

import cv2
import numpy as np

def data_augmentation(image_path, save_path):
    # 读取图像
    image = cv2.imread(image_path)
    
    # 图像翻转
    flipped_image = cv2.flip(image, 1)  # 1表示水平翻转，0表示垂直翻转
    
    # 图像旋转
    rows, cols = image.shape[:2]
    M = cv2.getRotationMatrix2D((cols/2, rows/2), 45, 1)  # 旋转45度
    rotated_image = cv2.warpAffine(image, M, (cols, rows))
    
    # 图像缩放
    scaled_image = cv2.resize(image, None, fx=0.8, fy=0.8)  # 缩小为原来的80%
    
    # 图像亮度调整
    brightness_image = cv2.convertScaleAbs(image, alpha=1.2, beta=10)  # 增加亮度
    
    # 保存增强后的图像
    cv2.imwrite(save_path + '/flipped.jpg', flipped_image)
    cv2.imwrite(save_path + '/rotated.jpg', rotated_image)
    cv2.imwrite(save_path + '/scaled.jpg', scaled_image)
    cv2.imwrite(save_path + '/brightness.jpg', brightness_image)

# 测试代码
data_augmentation('original.jpg', 'augmented')

这个代码会读取名为 `original.jpg` 的图像，并对其进行翻转、旋转、缩放和亮度调整等增强操作，然后将增强后的图像保存到指定的文件夹中。你可以根据自己的需求修改函数中的增强方法和参数。记得安装 OpenCV 库：`pip install opencv-python`。