python读取xml文件并修改object目标label的信息

"""
python 读取xml文件内容并完成修改
"""
import os
import xml.etree.ElementTree as ET
from xml.dom.minidom import parse# 转换xml文件
def bboxes2xml(folder, img_name, width, height, gts, xml_save_to):xml_file = open((xml_save_to + '/' + img_name + '.xml'), 'w')xml_file.write('<annotation>\n')xml_file.write('    <folder>' + folder + '</folder>\n')xml_file.write('    <filename>' + str(img_name) + '.jpg' + '</filename>\n')xml_file.write('    <size>\n')xml_file.write('        <width>' + str(width) + '</width>\n')xml_file.write('        <height>' + str(height) + '</height>\n')xml_file.write('        <depth>3</depth>\n')xml_file.write('    </size>\n')for gt in gts:xml_file.write('    <object>\n')xml_file.write('        <name>' + str(gt[0]) + '</name>\n')xml_file.write('        <pose>Unspecified</pose>\n')xml_file.write('        <truncated>0</truncated>\n')xml_file.write('        <difficult>0</difficult>\n')xml_file.write('        <bndbox>\n')xml_file.write('            <xmin>' + str(gt[1]) + '</xmin>\n')xml_file.write('            <ymin>' + str(gt[2]) + '</ymin>\n')xml_file.write('            <xmax>' + str(gt[3]) + '</xmax>\n')xml_file.write('            <ymax>' + str(gt[4]) + '</ymax>\n')xml_file.write('        </bndbox>\n')xml_file.write('    </object>\n')xml_file.write('</annotation>')xml_file.close()# 方式一读取xml文件
def byfinall(inputpath,xml_save_to):listdir = os.listdir(inputpath)for file in listdir:file_name=fileif file.endswith("xml"):file = os.path.join(inputpath, file)        # 拼接完整路径tree = ET.parse(file)                       # 加载xml文件,获取xml树对象root = tree.getroot()                       # 获取根节点tag_name = root.tag                         # .tag 得到标签的名称print("输出根节点的名称：",tag_name)size_dot=root.findall("size")   # 得到size节点width_=size_dot[0].findall("width")[0].text   # 读取xml获得图像宽度height_ = size_dot[0].findall("height")[0].text  # 读取xml获得图像高度depth_=size_dot[0].findall("depth")[0].text      # # 读取xml获得图像深度print(width_,height_,depth_)print("**"*10)gts=[]   # 列表用于存放标签、xmim,for object in root.findall("object"):name_obj = object.findall("name")       # 返回的是一个listname_text = name_obj[0].text            # 得到name文本if name_text in ['Knife']:name_text='knife'print("%s类别的坐标为：..."%(name_text))# print("%s类的位置坐标为。。。。" % (name_objs[0].text))single_obj=[]single_obj.append(name_text)bndbox_object=object.find("bndbox")xmin = bndbox_object.find('xmin').textxmax = bndbox_object.find('xmax').textymin = bndbox_object.find('ymin').textymax = bndbox_object.find('ymax').textsingle_obj.extend([xmin,ymin,xmax,ymax])gts.append(single_obj)print("xmin:",xmin,"ymin:",ymin,"xmax:",xmax,"ymax:",ymax)folder='images'img_name=os.path.splitext(file_name)[0]width=width_height=height_bboxes2xml(folder, img_name, width, height, gts, xml_save_to)if __name__ == '__main__':"""knifecigarettegundownviolence"""# 方式一读取xml文件inputpath=r"C:\Users\GuoQingru\Downloads\sigara"xml_save_to = r'C:\Users\GuoQingru\Downloads\sigara\xml'byfinall(inputpath,xml_save_to)