三 数据集处理之修改xml文件的存放路径信息

# 修改xml文件的存放路径信息 成功

import xml.dom.minidom
import os

path = 'D:/xml-test/4'  # xml文件存放路径
sv_path = 'D:/xml-test/annotation'  # 修改后的xml文件存放路径
files = os.listdir(path)

for xmlFile in files:
    dom = xml.dom.minidom.parse(os.path.join(path, xmlFile))  # 打开xml文件，送到dom解析

    root = dom.documentElement  # 得到文档元素对象

    item = root.getElementsByTagName('path')
    # 获取path这一node名字及相关属性值

    a, b = os.path.splitext(xmlFile)  # 分离出文件名a

    for i in item:
        i.firstChild.data = a + '.jpg'
        # i.firstChild.data = "D:/xml-test/4/" + a + '.png'

    with open(os.path.join(sv_path, xmlFile), 'w') as fh:
        dom.writexml(fh)