写一个python程序数据增强反转对称

在Python中，数据增强是一种技术，用于扩充训练数据集，通常应用于图像处理中，例如图像识别、计算机视觉等任务。其中反转和对称操作是很常见的增强手段。这里是一个简单的例子，展示如何使用PIL（Python Imaging Library）库来实现图片的水平翻转（左右镜像）和垂直翻转（上下镜像）：

from PIL import Image

def data_augmentation(image_path):
    # 打开图片
    img = Image.open(image_path)
    
    # 反转操作
    # 水平翻转
    flipped_img_h = img.transpose(Image.FLIP_LEFT_RIGHT)
    # 垂直翻转
    flipped_img_v = img.transpose(Image.FLIP_TOP_BOTTOM)

    # 保存增强后的图片
    flipped_img_h.save("horizontal_flip_" + image_path)
    flipped_img_v.save("vertical_flip_" + image_path)

# 使用函数并指定原图路径
data_augmentation('original_image.jpg')

这个程序会创建两个新的文件，分别是原图的水平翻转版本（horizontal_flip_...）和垂直翻转版本（vertical_flip_...）。你可以根据需要调整这个过程，并将其集成到你的机器学习模型的数据预处理流程中。