在多个voc数据合并在一起的时候,有时候会存在数据集标注的class分类不同,所以需要更换类别名称,此代码功能用于搜索需要更改的name。
import os
import xml.etree.ElementTree as ET
#程序功能:批量修改VOC数据集中xml标签文件的标签名称
def changelabelname(inputpath):
listdir = os.listdir(inputpath)
for file in listdir:
if file.endswith('xml'):
file = os.path.join(inputpath,file)
tree = ET.parse(file)
root = tree.getroot()
for object1 in root.findall('object'):
for sku in object1.findall('name'): #查找需要修改的名称
if (sku.text == 'person'): #‘preName’为修改前的名称
sku.text = 'headshoulder' #‘TESTNAME’为修改后的名称
print('更新成功')
tree.write(file,encoding='utf-8') #写进原始的xml文件并避免原始xml中文字符乱码
if __name__ == '__main__':
inputpath = 'C:\\Users\\Administrator\\Desktop\\Annotations_person\\' #此处替换为自己的路径
changelabelname(inputpath)