数据增广——旋转、镜像

1. 镜像

from PIL import Image  
import os  
  
## 图像镜像处理函数  
def mirror_image(image_path, output_path):  
    image = Image.open(image_path)  
    flipped_image = image.transpose(Image.FLIP_LEFT_RIGHT)  
    flipped_image.save(output_path)  
  
## 定义输入和输出文件夹路径  
input_dir = 'file1'  
output_dir = 'file2'  
   
  
## 遍历输入文件夹中的所有图像  
for filename in os.listdir(input_dir):  
    if filename.endswith('.jpg') or filename.endswith('.png'):   
        ## 构建输入和输出文件路径  
        input_path = os.path.join(input_dir, filename)  
        output_path = os.path.join(output_dir, filename)  
        ## 进行镜像处理  
        mirror_image(input_path, output_path)

2. 旋转

from PIL import Image  
import os  
  
## 设置输入和输出文件路径  
input_dir = 'file1'  
output_dir = 'file2'  
  
## 遍历输入目录中的所有图像文件  
for filename in os.listdir(input_dir):  
    ## 构建输入和输出文件路径  
    input_path = os.path.join(input_dir, filename)  
    output_path = os.path.join(output_dir, filename)  
      
    ## 打开图像并旋转90度  
    with Image.open(input_path) as img:  
        rotated_img = img.rotate(-90)  
          
    ## 保存旋转后的图像到输出文件路径  
    rotated_img.save(output_path)