python XML文件读写解析

在目标检测中，常常要用到labelimg、labelme等标注软件。打标时往往需要打开xml文件，下面介绍利用python解析xml文件的方法。

一、读xml文件

xml文件一般格式如下：

<?xml version="1.0"?>

-<annotation>

<folder>VOC2007</folder>

<filename>d10_1597367340_img_2.jpg</filename>

<path>C:123/d10_1597367340_img_2.jpg</path>


-<source>

<database>Unknown</database>

</source>


-<size>

<width>1920</width>

<height>1080</height>

<depth>3</depth>

</size>

<segmented>0</segmented>


-<object>

<name>pig</name>

<pose>0</pose>

<truncated>0</truncated>

<difficult>0</difficult>


-<bndbox>

<xmin>860</xmin>

<ymin>733</ymin>

<xmax>1189</xmax>

<ymax>910</ymax>

</bndbox>

        XML是可扩展标记语言 ，标记数据、定义数据类型，是一种允许用户对自己的标记语言进行定义的源语言。其中有且只有一个根节点，如上例中根节点为annotations,<></>是一个标签对，标签对内可以有该节点的子节点及其标签对；标签对的内容是可以自定义的。

       在用于标注数据的xml文件中，如一般根节点为annotations，包含size、folder、filename、object等多个子节点，object子结点中为bbox的信息。

xml文件读取方法：

import xml.etree.ElementTree as ET

ET为解析xml文件的专用库。

dom=ET.parse(xmlpath+item)
root=dom.getroot()
allobj=root.findall("object")

ET.parse()将xml文件读入到dom,返回一个etree对象，可以通过etree的getroot()、find()等函数对树的根节点和某个子节点进行访问。如findall("object")则返回所有的object节点，还可以通过.text()访问节点的文本属性。

二、写xml文件

写入xml文件同样是通过ET完成，

ET中的ET.Element（）为创建根节点，

folder=ET.SubElement(root, "folder")则为根节点root创建子节点folder,变量名为folder

可以直接通过folder.text=str(filename)对节点内容进行赋值

tree = ET.ElementTree(root)赋值完成后通过ElementTree将所有节点返回成Tree格式

然后就可以通过tree.write(filename, xml_declaration=False, encoding='utf-8')进行保存。

下面给出创建xml文件的代码，代码中已经将写入xml定义成了类对象，可以方便的调用

class GenAnnotations:
    def __init__(self, filename, path,witdh, height, depth=3, foldername="VOC2007", database = "Unknown",seg = 0):
        self.root = ET.Element("annotation")
        self.foleder = ET.SubElement(self.root, "folder")
        self.filename = ET.SubElement(self.root, "filename")
        self.path = ET.SubElement(self.root,"path")
        self.source = ET.SubElement(self.root, "source")
        self.database = ET.SubElement(self.source,"database")
        self.size = ET.SubElement(self.root, "size")
        self.width = ET.SubElement(self.size, "width")
        self.height = ET.SubElement(self.size, "height")
        self.depth = ET.SubElement(self.size, "depth")
        self.segmented = ET.SubElement(self.root,"segmented")

        self.foleder.text = foldername
        self.filename.text = filename
        self.path.text = path
        self.database.text = database
        self.width.text = str(witdh)
        self.height.text = str(height)
        self.depth.text = str(depth)
        self.segmented.text = str(seg)

    def savefile(self, filename):
        tree = ET.ElementTree(self.root)
        tree.write(filename, xml_declaration=False, encoding='utf-8')

    def add_object(self, label, xmin, ymin, xmax, ymax, tpose=0, ttruncated=0, tdifficult=0):
        object = ET.SubElement(self.root, "object")
        namen = ET.SubElement(object, "name")
        namen.text = label
        pose = ET.SubElement(object, "pose")
        pose.text = str(tpose)
        truncated = ET.SubElement(object, "truncated")
        truncated.text = str(ttruncated)
        difficult = ET.SubElement(object, "difficult")
        difficult.text = str(tdifficult)
        bndbox = ET.SubElement(object, "bndbox")
        xminn = ET.SubElement(bndbox, "xmin")
        xminn.text = str(xmin)
        yminn = ET.SubElement(bndbox, "ymin")
        yminn.text = str(ymin)
        xmaxn = ET.SubElement(bndbox, "xmax")
        xmaxn.text = str(xmax)
        ymaxn = ET.SubElement(bndbox, "ymax")
        ymaxn.text = str(ymax)


if __name__ == '__main__':
  

   anno= GenAnnotations(filename,path, 1920, 1080)
   with open (txtpath+item,'r',encoding='utf-8') as files:
###给出要写入的参数，可以从其他格式的标签中直接读取，pig为目标类别
        anno.add_object("pig",xmin,ymin,xmax,ymax)
        anno.savefile(xmlpath+item.split('.')[0]+'.xml')