# -*- coding:utf-8 -*-
# @Author len
# @Create 2023/10/19 10:11

from PIL import Image
import os
import glob

def convert_png_to_jpg(png_folder, jpg_folder):
    # 确保目标文件夹存在
    if not os.path.exists(jpg_folder):
        os.makedirs(jpg_folder)

    # 获取文件夹中所有PNG图像文件的路径
    png_files = glob.glob(os.path.join(png_folder, '*.png'))

    for png_file in png_files:
        try:
            # 获取PNG文件名（不包括文件路径和扩展名）
            file_name = os.path.splitext(os.path.basename(png_file))[0]

            # 打开PNG图像文件
            png_image = Image.open(png_file)

            # 将PNG图像转换为JPG格式
            jpg_image = png_image.convert("RGB")

            # 构造目标JPG文件路径
            jpg_file = os.path.join(jpg_folder, file_name + '.jpg')

            # 保存为JPG文件
            jpg_image.save(jpg_file, "JPEG")
            print(f"转换成功：{png_file} -> {jpg_file}")
        except Exception as e:
            print(f"转换失败：{png_file}，错误信息：{str(e)}")

# 指定源文件夹路径和目标文件夹路径
png_folder = r"D:\Waste\嵌入式\数据集\二维码\len\1"
jpg_folder = r"D:\Waste\嵌入式\数据集\二维码\len\1"

# 调用函数进行批量转换
convert_png_to_jpg(png_folder, jpg_folder)