数据增强：水平与垂直翻转

import os
import cv2

# 待增强图像所在目录
img_dir = r'C:\Users\yewenjing\Desktop\PF\new_64PF_img - 副本'

def horizontal_flip(image):
    """水平翻转图像"""
    return cv2.flip(image, 0)

# 读取所有图像文件名
img_files = os.listdir(img_dir)

# 对每张图像进行水平翻转，并保存到新文件
for img_file in img_files:
    # 读取原始图像
    image = cv2.imread(os.path.join(img_dir, img_file))
    # 进行水平翻转数据增强
    augmented_image = horizontal_flip(image)
    # 保存增强后的图像
    augmented_file = os.path.join(img_dir, 'flip_01'+img_file)
    cv2.imwrite(augmented_file, augmented_image)