智能车创意组地平线赛道--Yolov5数据集标注xml文件时统一标签问题

废话不多说，直接给出代码。

'''
通过解析xml文件，批量修改xml文件里的标签名称，比如把标签zero改成num
'''
import os.path
import glob
import xml.etree.ElementTree as ET

path = r'D:\test'    #存储标签的路径，修改为自己的Annotations标签路径
for xml_file in glob.glob(path + '/*.json'):
    ####### 返回解析树
	tree = ET.parse(xml_file)
	##########获取根节点
	root = tree.getroot()
	#######对所有目标进行解析
	for member in root.findall('object'):
		objectname = member.find('name').text
		if objectname == 'spontoon_010.jpg':      #原来的标签名字
			print(objectname)
			member.find('name').text = str('spontoon_010')    #替换的标签名字
			tree.write(xml_file)

这边是修改原始的标签名和替换的标签名，略显繁琐，如果有更好的方法，也欢迎各位能提出来。