(目标检测适用)批量修改xml文件中的name字段

tags: Python

Change ‘name’ in xml files.

在目标检测任务中，需要自己制作标签数据集。有时需要修改目标的类别名称，或者将某一类别名称修改为另一名称，或者将某些类别名称合并为新的名称。

import os
import os.path
from xml.etree.ElementTree import parse, Element


# 将某一类别的名称改成另一个名称
def changeName(xml_fold, origin_name, new_name):
    '''
    xml_fold: xml存放文件夹
    origin_name: 原始名字，比如弄错的名字，原先要cow,不小心打成cwo
    new_name: 需要改成的正确的名字，在上个例子中就是cow
    '''
    files = os.listdir(xml_fold)
    cnt = 0 
    for xmlFile in files:
        file_path = os.path.join(xml_fold, xmlFile)
        dom = parse(file_path)
        root = dom.getroot()
        for obj in root.iter('object'):#获取object节点中的name子节点