labelimage使用过程bug

在通过labelimage使用yolo格式进行png格式图片标注时会出现闪退情况

可将png格式转换为jpg,即可解决问题。

import os
import shutil

png_path = "C:/Users/a1566/Desktop/hs1/data/"  # png格式图片所在文件夹的路径
jpg_path = "C:/Users/a1566/Desktop/hs1/hsjpg/"  # 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请按回车键退出!")    # 暂停查看信息

猜你喜欢

转载自blog.csdn.net/qq_45014374/article/details/126503930