目标检测xml标签名修改

原因

目标检测的数据标注中,很多标签名标注错误,需要修改

代码

import glob
import xml.etree.ElementTree as ET
from tqdm import tqdm

def change_xml(xml):
    doc = ET.parse(xml)
    root = doc.getroot()
    objs = root.findall('object')
    for obj in objs:
        name = obj.find('name')
        if name.text == 'sss':
            name.text = 'bbb'
        elif name.text == 'www':
            name.text = 'gggg'
        elif name.text == 'hh':
            name.text = 'hhhh'
    doc.write(xml)


if __name__ == '__main__':
    xml_path = r'C:\data\Annotations\*.xml'
    xml_path_list = glob.glob(xml_path)
    for xml_filepath in tqdm(xml_path_list):
        change_xml(xml_filepath)