import os
from PIL import Image

def convert_jpg_to_png(directory):
    # 递归遍历目录中的所有 .jpg 文件
    for root, _, files in os.walk(directory):
        for file in files:
            if file.lower().endswith(".jpg"):
                jpg_path = os.path.join(root, file)
                png_path = os.path.splitext(jpg_path)[0] + ".png"
                try:
                    # 打开并转换图片
                    img = Image.open(jpg_path)
                    img.save(png_path)
                    print(f"Converted {jpg_path} to {png_path}")
                    
                    # 删除原始的 .jpg 文件
                    os.remove(jpg_path)
                    print(f"Deleted original file: {jpg_path}")
                except Exception as e:
                    print(f"Error converting {jpg_path}: {e}")

# 调用函数
directory = "/mnt/CVPR/diffusion-upo/generation/"
convert_jpg_to_png(directory)
directory = "/mnt/CVPR/diffusion-upo/generation_hpsv2/"
convert_jpg_to_png(directory)
directory = "/mnt/CVPR/diffusion-upo/generation_parti/"
convert_jpg_to_png(directory)