xml批量修改类别，修改后支持导入labelimg复核

import os
import xml.etree.ElementTree as ET
from tqdm import tqdm
import shutil

input_xml_path = r'H:\shaobing0424'
output = r"H:\shaobing0424_out"

dict_data = {"bike":"fff","ff":"kk"} # {需要修改的类别：修改成什么类别,.....}

for xml_root, xml_dirs, xml_files in os.walk(input_xml_path):
    for xml_filename in tqdm([x for x in xml_files if x.endswith('.xml')]):
        xml_path = os.path.join(xml_root,xml_filename)
        tree = ET.parse(xml_path)
        root = tree.getroot()
        for member in root.findall('object'):
            objectname = member.find('name').text
            if objectname in  dict_data.keys():
                member.find('name').text = dict_data[objectname]
                tree.write(os.path.join(output,xml_filename))
        if os.path.exists(os.path.join(output,xml_filename)) == False:
            shutil.copy(os.path.join(xml_root,xml_filename),os.path.join(output,xml_filename))
        try:

            shutil.move(xml_path.replace(".xml",".jpg"),output)
        except:
            shutil.move(xml_path.replace(".xml", ".png"),output)