labelimage标注label批量修改标签

import os
import xml.etree.ElementTree as ET

def modify_xml_labels(xml_folder):
    modified_files = []
    for root, dirs, files in os.walk(xml_folder):
        for file in files:
            if file.endswith(".xml"):
                xml_path = os.path.join(root, file)
                tree = ET.parse(xml_path)
                root_elem = tree.getroot()
                modified = False
                for obj in root_elem.findall("object"):
                    name = obj.find("name")
                    if name.text == "faultd":
                        name.text = "fault"
                        modified = True
                    elif name.text == "nohelmet_manned":
                        name.text = "manned_without_helmet"
                        modified = True
                    elif name.text == "helmet_manned":
                        name.text &

你可能感兴趣的:(python)