对labelme标注的数据进行增强后,需要批量将label_json文件夹中的label.png重名为对应的文件名,该如何处理?
我在网上找到了一段代码,但运行结果有误,例如文件名为taoguan_000,运行后label.png变为taoguan.png,而不是taoguan_000.png
# 把label.png改名为原图名.png
import os
for root, dirs, names in os.walk("D:/labelme_json"): # 改成你自己的label_json文件夹所在的目录
for dr in dirs:
file_dir = os.path.join(root, dr)
# print(dr)
file = os.path.join(file_dir, 'label.png')
# print(file)
new_name = dr.split('_')[0] + '.png'
new_file_name = os.path.join(file_dir, new_name)
os.rename(file, new_file_name)