关于VOC数据集迁移后path路径修改的脚本

##修改xml文件中的<path>，与实际数据路径对应
##author:leo

# coding=utf-8
import os
import os.path
import xml.dom.minidom

file_path_pic = "you--picture--path/JPEGImages"
file_path = "you——xml--path/train"
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')
        # 原始信息
        #print(original_path)
        p0=original_path[0]
        #print(p0)
        path0=p0.firstChild.data   #原始路径
        #print(path0)
        # 修改
        jpg_name=path0.split('\\')[-1] #获取图片名
        #print(jpg_name)
        modify_path=file_path_pic+"/"+jpg_name   #修改后path
        print(modify_path)
        #p0.firstChild.data=modify_path

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