Python批量修改.xml文件里的路径

在做目标检测的时候，下载的或者是别人的数据集里的.xml文件的路径，需要修改为自己的图片的保存的地址，比如：别人的数据集的图片保存路径为：D:\PyCharmAnaconda\BoltDetection\JPEGImage\000001.jpg通过Python脚本实现之后图片路径正确改变了修改后的路径为：E:/Data_study/yolo_tf2.1/VOCdevkit/VOC2007/Anno

逆风飞扬fei

1549人浏览 · 2020-06-11 08:25:50

逆风飞扬fei · 2020-06-11 08:25:50 发布

在做目标检测的时候，下载的或者是别人的数据集里的.xml文件的路径，需要修改为自己的图片的保存的地址，比如：
在这里插入图片描述
别人的数据集的图片保存路径为：
D:\PyCharmAnaconda\BoltDetection\JPEGImage\000001.jpg
通过Python脚本实现之后图片路径正确改变了

在这里插入图片描述
修改后的路径为：E:/Data_study/yolo_tf2.1/VOCdevkit/VOC2007/Annotations/000001.jpg

下面说说代码实现：

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

# .xml文件地址
path = "E:/Data_study/yolo_tf2.1/VOCdevkit/VOC2007/Annotations/"
# 得到文件夹下所有文件名称
files = os.listdir(path)  
s = []
# 遍历文件夹
for xmlFile in files:
    # 判断是否是文件夹,不是文件夹才打开
    if not os.path.isdir(xmlFile):
        print(xmlFile)
        pass
    path = "E:/Data_study/yolo_tf2.1/VOCdevkit/VOC2007/Annotations/"
    newStr = os.path.join(path, xmlFile)
    #最核心的部分,路径拼接,输入的是具体路径
    #得到.xml文件的根（也就是annotation）
    dom = parse(newStr)
    root = dom.getroot()
    #获得后缀.前的文件名(分离文件名和扩展名)
    part = os.path.splitext(xmlFile)[0]
    # 文件名+后缀
    part1 = part + '.jpg'
    # path里的新属性值：
    newStr1 = 'E:/keras-yolo3-master/VOCdevkit/VOC2007/JPEGImages/' + part1
    #通过句柄找到path的子节点，然后给子节点设置内容
    root.find('path').text = newStr1  
    # #打印输出
    print('已经修改')
    dom.write(newStr, xml_declaration=True)
    pass