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

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

Python批量修改.xml文件里的路径_第2张图片
修改后的路径为: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

你可能感兴趣的:(一些Python小脚本)