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)
对数据集标注时容易出现名字很乱的情况,有时训练时会出错,这时就需要对文件重新命名,代码附上。