Python スクリプト-----異なるカテゴリの XML ファイルをマージし、名前を変更します

1. 異なるカテゴリの XML ファイルを結合する

import cv2
import numpy as np
import os
import xml.etree.ElementTree as ET
from xml.dom import minidom
import matplotlib.pyplot as plt

CLASSES_NAMES=['C1','W','G1','S1','S2', 'LC1']

def get_class_names(CLASSES_NAMES):
    """
    return a dict, as following,
        {'pottedplant': 0,
         'sheep': 1,
         'bird': 2,
         'dog': 3,
         'bottle': 4,
         'boat': 5,
         ......
        }
    """
    object_category2id = {
   
    
    }
    for i, item in enumerate(CLASSES_NAMES):
        object_category2id[item] = i
    return object_category2id

#获取类别对应标签id，输出为字典，通过类别名称获取id，如：cname2cid[‘bird’]=2
cname2cid=get_class_names(CLASSES_NAMES)

def write_xml():
    annotation = ET.Element("annotation")
    filename = ET.Element("filename")
    filename.text='CRH2A-2309-14-22.jpg'
    size=ET.Element("size")
    width=ET.Element("width")
    width.text=str(2000)
    height=ET.Element("height")
    height.text=str(2048)
    depth = ET.Element("depth")
    depth.text = str(3)
    size.extend([width,height,depth])
    annotation.extend([filename,size])
    #=======object_start=======
    #=====循环这部分代码加入不同目标的标注信息========
    #===需要更改的变量：name xmin ymin xmax ymax
    object=ET.Element("object")
    name=ET.Element("name")
    name.text='class_1'
    pose=ET.Element("pose")
    pose.text='Unspecified'
    truncated=ET.Element("truncated")
    truncated.text=str(0)
    difficult=ET.Element("difficult")
    difficult.text=str(0)
    bndbox=ET.Element("bndbox")
    xmin=ET.Element("xmin")
    xmin.text=str(1147)
    ymin=ET.Element("ymin")
    ymin.text=str(1199)
    xmax=ET.Element("xmax")
    xmax.text=str(1503)
    ymax=ET.Element("ymax")
    ymax.text=str(1617)
    bndbox.extend([xmin,ymin,xmax,ymax])
    object.extend([name, pose, truncated, difficult,bndbox])
    #=======object_end=======
    annotation.append(object)
    # tree = ET.ElementTree(annotation)
    # tree.write("output.xml", encoding="utf-8", xml_declaration=True)  #保存时无缩进，添加缩进需要借用dom


    #借用dom，添加缩进
    rawtext = ET.tostring(annotation)
    dom = minidom.parseString(rawtext)
    with open("output.xml", "w") as f:
        dom.writexml(f, indent="",addindent="  ", newl="\n", encoding="utf-8")

    # tree = ET.ElementTree(annotation)
    # tree.write("output.xml", encoding="utf-8", xml_declaration=True)  #保存时无缩进，添加缩进需要借用dom


    #借用dom，添加缩进
    rawtext = ET.tostring(annotation)
    dom = minidom.parseString(rawtext)
    with open("output.xml", "w") as f:
        dom.writexml(f, indent="",addindent="  ", newl="\n", encoding="utf-8")

def WriteObjectsToXml(ObjDict, save_path):


    annotation = ET.Element("annotation")
    filename = ET.Element("filename")
    filename.text = ObjDict