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

from PIL import Image
import os

# 定义函数，旋转图片并保存
def rotate_images(input_dir, output_dir, angle):
    for filename in os.listdir(input_dir):
        # 判断文件是否为图片
        if filename.endswith('.jpg') or filename.endswith('.png'):
            # 打开图片
            img = Image.open(os.path.join(input_dir, filename))
            # 旋转图片,使用expand=True参数来确保旋转后的图像不会被裁剪，若要裁剪图像，可以将此参数设置为False。
            rotated_img = img.rotate(angle, expand=True)
            # 保存旋转后的图片
            rotated_img.save(os.path.join(output_dir, filename))

# 输入文件夹路径、输出文件夹路径、旋转角度
input_dir = 'D:\\images'
output_dir = 'D:\\images_out'
angle = 345  # 逆时针旋转度数

rotate_images(input_dir, output_dir, angle)