python 批量修改文件名和图片xml内的path、folder

修改文件的:

#批量修改文件名
#批量修改图片文件名
import os
import re
import sys
def renameall():
	fileList = os.listdir(r"C:\Users\83543\Desktop\bytrain\beerbottle")		#待修改文件夹
	print("修改前:"+str(fileList))		#输出文件夹中包含的文件
	currentpath = os.getcwd()		#得到进程当前工作目录
	os.chdir(r"C:\Users\83543\Desktop\bytrain\beerbottle")		#将当前工作目录修改为待修改文件夹的位置
	num=1		#名称变量
	for fileName in fileList:		#遍历文件夹中所有文件
		pat=".+\.(jpg|png|gif)"		#匹配文件名正则表达式
		pattern = re.findall(pat,fileName)		#进行匹配
		os.rename(fileName,("beerbottle"+str(num)+'.'+pattern[0]))		#文件重新命名
		num = num+1		#改变编号,继续下一项
	print("---------------------------------------------------")
	os.chdir(currentpath)		#改回程序运行前的工作目录
	sys.stdin.flush()		#刷新
	print("修改后:"+str(os.listdir(r"C:\Users\83543\Desktop\bytrain\beerbottle")))		#输出修改后文件夹中包含的文件
renameall()

修改xml的path的:

import xml.dom.minidom
import os

path = r'C:/Users/83543/Desktop/keras-yolo3-master/VOCdevkit/VOC2007/Annotations/'  # xml文件存放路径
sv_path = r'C:/Users/83543/Desktop/xml'  # 修改后的xml文件存放路径
files = os.listdir(path)
cnt = 1  ##从1开始计数,如果要在前面+0,用zfill函数

for xmlFile in files:
    dom = xml.dom.minidom.parse(os.path.join(path, xmlFile))  # 打开xml文件,送到dom解析
    root = dom.documentElement  # 得到文档元素对象
    item = root.getElementsByTagName('path')  # 获取path这一node名字及相关属性值
    print(item)
    for i in item:
        i.firstChild.data = 'C:\\Users\\83543\\Desktop\\keras-yolo3-master\\VOCdevkit\\JPEGImages'+'glass' + str(cnt) + '.jpg'  # xml文件对应的图片路径

    with open(os.path.join(sv_path, xmlFile), 'w') as fh:
        dom.writexml(fh)
    cnt += 1

修改folder:

跟上面的差不多。

import xml.dom.minidom
import os

path = r'C:/Users/83543/Desktop/keras-yolo3-master/VOCdevkit/VOC2007/Annotations/'  # xml文件存放路径
sv_path = r'C:/Users/83543/Desktop/xml'  # 修改后的xml文件存放路径
files = os.listdir(path)
cnt = 1  ##从1开始计数,如果要在前面+0,用zfill函数

for xmlFile in files:
    dom = xml.dom.minidom.parse(os.path.join(path, xmlFile))  # 打开xml文件,送到dom解析
    root = dom.documentElement  # 得到文档元素对象
    item = root.getElementsByTagName('folder')  # 获取path这一node名字及相关属性值
    print(item)
    for i in item:
        i.firstChild.data = 'glass'  # xml文件对应的图片路径

    with open(os.path.join(sv_path, xmlFile), 'w') as fh:
        dom.writexml(fh)
    cnt += 1

修改path和folder合在一起的

import xml.dom.minidom
import os

path = r'C:/Users/83543/Desktop/keras-yolo3-master/VOCdevkit/VOC2007/Annotations/'  # xml文件存放路径
sv_path = r'C:/Users/83543/Desktop/xml'  # 修改后的xml文件存放路径
files = os.listdir(path)
cnt = 1  ##从1开始计数,如果要在前面+0,用zfill函数

for xmlFile in files:
    dom = xml.dom.minidom.parse(os.path.join(path, xmlFile))  # 打开xml文件,送到dom解析
    root = dom.documentElement  # 得到文档元素对象
    item = root.getElementsByTagName('path')  # 获取path这一node名字及相关属性值
    item2 = root.getElementsByTagName('folder')  # 获取path这一node名字及相关属性值
    print(item)
    for i in item:
        i.firstChild.data = 'C:\\Users\\83543\\Desktop\\keras-yolo3-master\\VOCdevkit\\JPEGImages'+'glass' + str(cnt) + '.jpg'  # xml文件对应的图片路径

    for i in item2:
        i.firstChild.data = 'glass' ##folder
    with open(os.path.join(sv_path, xmlFile), 'w') as fh:
        dom.writexml(fh)
    cnt += 1

你可能感兴趣的:(python)