【Tool】labelimg 标注合并

前言

当团队标注时，为了方便，让每人只标注所有图片的其中一个类别，最后在对每个人的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}")