PAC VOC数据集annotations文件夹xml文件path路径修改为对应图片路径代码

Python代码实现

# coding=utf-8
import os
import os.path
import xml.dom.minidom
# 替换xml文件存放路径
file_path = "/home/hq/Research/data/VOC/VOCtrainval-2012/VOCdevkit/VOC2012/Annotations"
# 替换图片保存路径
picture_path = "/home/hq/Research/data/VOC/VOCtrainval-2012/VOCdevkit/VOC2012/JPEGImages"
files = os.listdir(file_path)  # 得到文件夹下所有文件名称
s = []
for xmlFile in files:  # 遍历文件夹
    if not os.path.isdir(xmlFile):  # 判断是否是文件夹,不是文件夹才打开
        print(xmlFile)
        # xml文件读取操作
        # 将获取的xml文件名送入到dom解析
        dom = xml.dom.minidom.parse(os.path.join(file_path, xmlFile))  ###最核心的部分,路径拼接,输入的是具体路径
        root = dom.documentElement
        # 获取标签对path之间的值
        original_path = root.getElementsByTagName('path')
        # 原始信息
        p0=original_path[0]
        path0=p0.firstChild.data   #原始路径
        print(path0)
        # 修改
        jpg_name=path0.split('\')[-1]   #获取图片名
        modify_path=picture_path+'/'+jpg_name   #修改后path
        p0.firstChild.data=modify_path

        # 保存修改到xml文件中
        with open(os.path.join(file_path, xmlFile), 'w') as fh:
            dom.writexml(fh)
            print('修改path OK!')