YOLO目标检测，将png格式图像转成jpg格式

介绍

• YOLOv8目标检测模型训练时，png格式图片和jpg格式图片不能混合在一起训练，这可能会影响模型最终训练的精度。
• 所以使用PIL库将png格式图片转成jpg格式图片。

Python实现

from PIL import Image
import os

"""
将png格式转成jpg格式
"""
def convert_png_to_jpg(png_path, jpg_path):
    try:
        # 打开PNG图片
        image = Image.open(png_path)
        # 转换并保存为JPG格式
        image = image.convert('RGB')
        image.save(jpg_path, 'JPEG')
        print(f"Converted {png_path} to {jpg_path}")
    except Exception as e:
        print(f"Error converting {png_path}: {str(e)}")

def convert_all_png_to_jpg_in_directory(directory):
    for filename in os.listdir(directory):
        if filename.endswith('.png'):
            png_path = os.path.join(directory, filename)
            jpg_path = os.path.join(directory, filename[:-4] + '.jpg')  # 用同名但不同后缀的方式创建JPG文件
            convert_png_to_jpg(png_path, jpg_path)

if __name__ == '__main__':
    # 要转换的PNG图片所在的文件夹路径
    png_directory1 = r'E:\xxx\images\train'
    png_directory2 = r'E:\xxx\images\val'
    
    # 转换所有PNG图片为JPG格式
    convert_all_png_to_jpg_in_directory(png_directory1)
    convert_all_png_to_jpg_in_directory(png_directory2)

    print("转换成功！")

---

整理不易🚀🚀，关注和收藏后拿走📌📌欢迎留言🧐👋📣✨
快来关注我的公众号🔎AdaCoding 和 GitHub🔎 AdaCoding123