使用python创建、添加、修改xml格式的标注文件

在深度学习目标检测的所用到的数据集中，常会遇到xml格式的标注文件（例如VOC格式的数据集采用的标注文件就是xml格式的）。在我们自己建立xml标注文件或者是将其他类型的标注文件（例如txt、json）转化到xml格式时，就需要了解这方面的知识。

下面我将列出用python借助lxml来进行xml类型文件的创建、添加、修改功能的实现代码。

1.创建xml格式的标注文件

下面展示创建xml格式文件的代码及运行代码后生成的xml文件的内容效果图，这里我们先简单创立一个xml文件，设定根节点、为根节点添加几个子节点、并给节点起名、设定节点包含的文本text信息。

代码如下：

from lxml.etree import Element, tostring, parse
from lxml.etree import SubElement as subElement
def xml_construct(save_path,folder,filename,path,width=800,height=600,depth = 3,segmented=0):
    default_text = 'default'
    node_root = Element('annotation')  # 根节点

    node_folder = subElement(node_root, 'folder')  # 在节点下添加名为'folder'的子节点
    node_folder.text = folder  # 设定节点的文字

    node_filename = subElement(node_root, 'filename')
    node_filename.text = filename

    node_path = subElement(node_root, 'path')
    node_path.text = path

    node_size = subElement(node_root, 'size')
    node_size_width = subElement(node_size, 'width')
    node_size_width.text = '%s' % int(width)
    node_size_height = subElement(node_size, 'height')
    node_size_height.text = '%s' % int(height)
    node_size_depth = subElement(node_size, 'depth')
    node_size_depth.text = '%s' % int(depth)

    node_segmented = subElement(node_root, 'segmented')
    node_segmented.text = '%s' % int(segmented)

    xml = tostring(node_root, pretty_print=True) #将上面设定的一串节点信息导出
    with open(save_path,'wb') as f: #将节点信息写入到文件路径save_path中
        f.write(xml)

    return

#----------调用上面所写的函数进行试验,创建名为test.xml的xml文件----------
xml_construct('test.xml','test','test','test',width=1600,height=1200,)

运行以上代码结果为：

2.为已存在的xml文件进行添加节点

在这里我们展示如何对已存在的xml文件进行添加节点操作，我们接着在上一步骤中生成的test.xml上面来进行修改。

代码如下：
 

from lxml.etree import Element, tostring, parse
from lxml.etree import SubElement as subElement
def xml_add_object(xml_path,name,id,xmin,ymin,xmax,ymax,pose = 'Unspecified',truncated=0,difficult=0):
    tree = parse(xml_path)  # 读取xml文件
    node_root = tree.getroot()  # 找到xml文件的根节点
    node_object = subElement(node_root, 'object')  #在根节点node_root下面添加名为'object'的子节点
    node_object_name = subElement(node_object, 'name') #在根节点node_root的子节点object下面继续添加子节点object的子节点'name'
    node_object_name.text = name #设定该节点的文本text信息
    node_object_id = subElement(node_object, 'id')
    node_object_id.text = '%s' % int(id)
    node_object_pose = subElement(node_object, 'pose')
    node_object_pose.text = pose
    node_object_truncated = subElement(node_object, 'truncated')
    node_object_truncated.text = '%s' % int(truncated)
    node_object_difficult = subElement(node_object, 'difficult')
    node_object_difficult.text = '%s' % int(difficult)
    # object坐标
    node_bndbox = subElement(node_object, 'bndbox')
    node_xmin = subElement(node_bndbox, 'xmin')
    node_xmin.text = '%s' % int(xmin)
    node_ymin = subElement(node_bndbox, 'ymin')
    node_ymin.text = '%s' % int(ymin)
    node_xmax = subElement(node_bndbox, 'xmax')
    node_xmax.text = '%s' % int(xmax)
    node_ymax = subElement(node_bndbox, 'ymax')
    node_ymax.text = '%s' % int(ymax)

    xml = tostring(node_root, pretty_print=True) #将修改后的xml节点信息导出

    with open(xml_path,'wb') as f: #将修改后的xml节点信息覆盖掉修改前的
        f.write(xml)

    return

#----------试验,在test.xml中进行添加节点操作----------
xml_add_object(xml_path='test.xml',name='test_name',id=999,xmin=1,ymin=2,xmax=3,ymax=4,)

运行以上代码后原先的test.xml文件就被添加了相应的子节点，效果如下：

3.修改已存在的xml文件中特定节点的信息

如果我们只想的对xml文件中某些特定节点信息进行修改的话，可以进行一下代码操作（这里我们还是以上面的test.xml文件为操作对象，将其中的size节点下的width和height节点信息进行修改）：

from lxml.etree import Element, tostring, parse
from lxml.etree import SubElement as subElement
def modify_size(xml_path,size_width=800,size_height=600):
    tree = parse(xml_path)
    width = tree.xpath('//width')#这里返回的width是一个包含所有名称中带有"width"的节点的列表
    for width_i in width:
        width_i.text = '%s' % int(size_width) #这里直接写width_i.text = size_width会报错

    height = tree.xpath('//height')
    for height_i in height:
        height_i.text = '%s' % int(size_height)

    node_root = tree.getroot()
    xml = tostring(node_root, pretty_print=True)
    with open(xml_path,'wb') as f:
        f.write(xml)

    return

#----------试验----------
#以上面的test.xml文件为操作对象，将其中的size节点下的width和height节点信息进行修改
modify_size('test.xml',size_width=800,size_height=600)

效果如下：