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

最新推荐文章于 2024-08-14 23:30:40 发布

anthony-36

最新推荐文章于 2024-08-14 23:30:40 发布

阅读量1.2k

点赞数 3

分类专栏：数据集格式转换文章标签： python pytorch 深度学习

本文链接：https://blog.csdn.net/qq_46067306/article/details/126360863

版权

数据集格式转换专栏收录该内容

2 篇文章 0 订阅

订阅专栏

该博客讲述了如何使用Python将D盘中指定路径的PNG图片转换为JPG格式，并保存到新建或清空的jpg文件夹内。程序首先检查jpg文件夹，如果存在则清空，不存在则创建。然后遍历PNG文件，逐个复制并转换为JPG格式。最后，确认所有PNG文件都已转换并保存成功。

摘要由CSDN通过智能技术生成

将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请按回车键退出！")    # 暂停查看信息