对labelme标注的数据进行增强后，需要批量将label_json文件夹中的label.png重名为对应的文件名，该如何处理？

对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)

img

img

img

img

你把这个输出也print 出来，看看哪里有问题，然后缺少的地方加进去。
new_name = dr.split('_')[0] + '.png'
猜测这个输出是 taoguan.png ，没有把'_000'加进去。