前言
当团队标注时,为了方便,让每人只标注所有图片的其中一个类别,最后在对每个人的xml进行合并。
解决
import os import xml.etree.ElementTree as ET xmls1_path="" xmls2_path="" save_xmls_path="" xmls1=os.listdir(xmls1_path) xmls2=os.listdir(xmls2_path) for i in xmls1: if i in xmls2: print("合并xml") tree1 = ET.parse(os.path.join(xmls1_path,i)) tree2 = ET.parse(os.path.join(xmls2_path,i)) objects1=tree1.findall('object') objects2=tree2.findall('object') for j in objects2: tree1.getroot().append(j) tree1.write(os.path.join(save_xmls_path,i)) print(len(tree1.findall('object'))) else: print(f"不存在xml文件{i}")