深度学习-批量数据集修改标签.py

import os.path
import xml.dom.minidom

# 将"cardboard"都替换为："paper"
# path为需要替换标签的目标文件夹
path = 'D:/new_xml'
files = os.listdir(path)  # 得到文件夹下所有文件名称
s = []
print('------------开始替换标签名称！--------------')
for xmlFile in files:  # 遍历原标签文件夹

    if not os.path.isdir(xmlFile):  # 判断是否是文件夹，不是文件夹才打开
        dom = xml.dom.minidom.parse(os.path.join(path, xmlFile))
        root = dom.documentElement
        # 替换节点，除了name也可以替换为其他节点
        pathNode = root.getElementsByTagName('name')
        print(pathNode)
        print(len(pathNode))
        j = len(pathNode)
        for i in range(j):
            if pathNode[i].firstChild.data == "cardboard" :
                print("替换前的名称为：", pathNode[i].firstChild.data)
                pathNode[i].firstChild.data = "paper"
                print("i为:", i)
                print("替换后的名称为：", pathNode[i].firstChild.data)
                i = i + 1
                with open(os.path.join(path, xmlFile), 'w', encoding='utf8') as fh:
                    dom.writexml(fh)

print('------------标签名称替换成功！--------------')