使用python将文件夹中的图片批量镜像(扩充数据集)

from PIL import Image
import os

# 定义输入和输出文件夹路径
input_folder = 'D:\\images'
output_folder = 'D:\\images_out'

# 遍历输入文件夹中的所有图片文件
for file_name in os.listdir(input_folder):
    # 检查文件是否为图片文件
    if file_name.endswith('.jpg') or file_name.endswith('.png'):
        # 打开图片文件
        image = Image.open(os.path.join(input_folder, file_name))
        # 使用了transpose方法来进行图片的镜像操作,method=Image.FLIP_LEFT_RIGHT表示水平镜像
        mirrored_image = image.transpose(method=Image.FLIP_LEFT_RIGHT)
        # 生成输出文件路径
        output_path = os.path.join(output_folder, file_name)
        # 保存镜像后的图片
        mirrored_image.save(output_path)