# -*- coding:utf-8 -*-
# @Author len
# @Create 2023/11/9 11:36


import os
from PIL import Image

def convert_webp_to_jpg(webp_path, jpg_path):
    image = Image.open(webp_path)
    image.save(jpg_path, 'JPEG')

def convert_webp_to_jpg_batch(input_folder, output_folder):
    # 确保输出文件夹存在
    if not os.path.exists(output_folder):
        os.makedirs(output_folder)

    # 遍历输入文件夹中的所有文件
    for file_name in os.listdir(input_folder):
        if file_name.endswith('.webp'):
            # 构建WebP文件的完整路径
            webp_path = os.path.join(input_folder, file_name)

            # 构建目标JPEG文件的完整路径
            jpg_path = os.path.join(output_folder, os.path.splitext(file_name)[0] + '.jpg')

            # 执行WebP到JPEG的转换
            convert_webp_to_jpg(webp_path, jpg_path)

# 指定输入文件夹和输出文件夹的路径
input_folder = r'D:\Waste\嵌入式\数据集\交通标志\1225交通标志\背景2'
output_folder = r'D:\Waste\嵌入式\数据集\交通标志\1225交通标志\背景'

# 调用函数进行批量转换
convert_webp_to_jpg_batch(input_folder, output_folder)