数据集+标签重命名

import os

pic_path = r"F:\pzy\helmet_fu\all-labels"  #输入自己要重命名的文件路径
file_type = ".txt"     #输入自己要重命名的文件类型

def rename(pic_path, file_type):
    piclist = os.listdir(pic_path)  #os.listdir的返回值是一个列表，列表里面存储该path下面的子目录的名称
    total_num = len(piclist)   #获取列表长度

    i = 1
    for pic in piclist:
        if pic.endswith(file_type):  #endswith()判断后缀
            old_path = os.path.join(os.path.abspath(pic_path), pic)
            new_path = os.path.join(os.path.abspath(pic_path), '00'+format(str(i), '0>3s') + file_type)

            os.renames(old_path, new_path)
            print(u"把原图片命名格式：" + old_path + u"转换为新图片命名格式：" + new_path)
            # print "把原图片路径：%s,转换为新图片路径：%s" %(old_path,new_path)
            i = i + 1

    print("总共有" + str(total_num) + "张图片被重命名为:" "0001" + file_type + '~' + '00' + format(str(i - 1), '0>3') + file_type)


rename(pic_path, file_type)

对数据集标注时容易出现名字很乱的情况，有时训练时会出错，这时就需要对文件重新命名，代码附上。