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

Albert_xiong

发布于 2021-06-21 17:41:39

2.7K0

发布于 2021-06-21 17:41:39

文章被收录于专栏：Mybatis学习

在做目标检测的时候，下载的或者是别人的数据集里的.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

本文参与腾讯云自媒体同步曝光计划，分享自作者个人站点/博客。

原始发表：2020/06/11 ，如有侵权请联系 cloudcommunity@tencent.com 删除

python