小结-YOLOV5训练自己的数据集，有些数据集是png的解决办法

这次我在做项目时，数据集有些事png图像，训练时出现下面的信息

虽然这个并不影响最终的结果，但是看的不舒服

 png格式的图像通道不能直接更改为jpg格式，需要改变通道

from PIL import Image
import os

# 设置需要转换的图片目录
img_dir = r""

# 遍历目录下所有文件
for filename in os.listdir(img_dir):
    # 判断文件是否为png格式
    if filename.endswith(".png"):
        # 构造新的文件名，将后缀改为jpg
        new_filename = os.path.splitext(filename)[0] + ".jpg"
        # 打开图片
        img = Image.open(os.path.join(img_dir, filename))
        # 如果图片是RGBA模式，转换为RGB模式
        if img.mode == "RGBA":
            img = img.convert("RGB")
        # 如果图片是P模式，转换为RGB模式
        if img.mode == "P":
            img = img.convert("RGB")
        # 保存为新的jpg格式图片
        img.save(os.path.join(img_dir, new_filename))
        # 关闭图片
        img.close()
        # 删除原来的png格式图片
        os.remove(os.path.join(img_dir, filename))

只改变后缀，不改变前缀