数据集png文件转jpg文件（python）

将D:\pytorch\centernet-pytorch-main\centernet-pytorch-main\kittidatasets\kitti\JPEGImage位置内的png文件转换成jpg，保存到D:\pytorch\centernet-pytorch-main\centernet-pytorch-main\kittidatasets\kitti\jpg文件夹下，保存前会清楚jpg内的文件。

import os
import shutil

png_path = "D:\pytorch\centernet-pytorch-main\centernet-pytorch-main\kittidatasets\kitti\JPEGImages"  # png格式图片所在文件夹的路径
jpg_path = "D:\pytorch\centernet-pytorch-main\centernet-pytorch-main\kittidatasets\kitti\jpg"  # jpg格式图片存放文件夹的路径
file_walk = os.listdir(png_path)
fileNum = 0         # png文件夹下所有文件个数计数
png_fileNum = 0     # png文件夹下png图片个数计数

for filePath in file_walk:
    fileNum += 1
    protion = os.path.splitext(filePath)

    if protion[1].lower() == '.png':  # 判断文件后缀是否为png
        if png_fileNum == 0:  # 当png文件夹中有png图片
            # 判断是否存在jpg文件夹，存在则清空文件夹，不存在就建立文件夹
            if os.path.exists(jpg_path):
                shutil.rmtree(jpg_path)
                os.mkdir(jpg_path)
                print("jpg文件夹内原文件已清除")
            else:
                os.mkdir(jpg_path)
                print("jpg文件夹已创建")
        png_fileNum += 1
        print("正在处理：" + filePath)

        # 复制转存png图片为jpg格式到jpg文件夹中
        shutil.copyfile(os.path.join(png_path, filePath), os.path.join(jpg_path, protion[0] + '.jpg'))

print('\n文件夹内共有' + str(fileNum) + '个文件，其中png格式文件有' + str(png_fileNum) + '个，已全部完成转换，存至jpg文件夹内')
# stop = input("\n\n请按回车键退出！")    # 暂停查看信息