用Python写一段扩充voc数据集的代码

以下是Python代码，用于扩充voc数据集：

import os
from PIL import Image

# 设置输入输出路径
input_dir = '/path/to/input/dir'
output_dir = '/path/to/output/dir'

# 获取输入目录下的所有图像文件
image_files = [f for f in os.listdir(input_dir) if f.endswith('.jpg') or f.endswith('.jpeg') or f.endswith('.png')]

# 如果输出目录不存在，则创建输出目录
if not os.path.exists(output_dir):
    os.makedirs(output_dir)

for file in image_files:
    # 文件路径
    file_path = os.path.join(input_dir, file)

    # 打开图像文件
    with Image.open(file_path) as image:
        # 旋转图像，扩充数据集
        for angle in [0, 90, 180, 270]:
            rotated_image = image.rotate(angle)

            # 设置保存文件名和保存路径
            save_name = file.split('.')[0] + '_rotated_' + str(angle) + '.jpg'
            save_path = os.path.join(output_dir, save_name)

            # 保存图像文件
            rotated_image.save(save_path)

这段代码首先设置了输入和输出目录，然后获取输入目录下的所有图像文件。接着，对于每个图像文件，打开它并将其旋转90度，180度和270度，然后将旋转后的图像文件保存到输出目录。这样就实现了扩充voc数据集的目的。