python记录(4)- lxml模块创建xml文件

因为要比赛制作voc数据集，然后给的数据集label是一个.txt文件，然后就需要将这个txt文件变为xml文件。对这方面一点也不懂，在此记录一下用lxml模块中的 etree创建xml文件。这只是一个例子，并不是制作voc数据集。后面估计会更新voc数据集制作。

效果

  VOCtype
  000001.jpg
  
    VOC
  
  
    458
    45
    45
  
  0
  
  

代码

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
Created on Wed Apr 11 15:12:22 2018

@author: pc314
"""

from lxml import etree
#创建一个annotion节点
root = etree.Element('annotion')
#创建一个子节点folder，一定要指定父节点
child1 = etree.SubElement(root,'folder')
child1.text = 'VOCtype'

child2 = etree.SubElement(root,'filename')
child2.text = '000001.jpg'

child3 = etree.SubElement(root,'source')

child31 = etree.SubElement(child3,'database')
child31.text = 'VOC'

child4 = etree.SubElement(root,'size')


child41 = etree.SubElement(child4,'width')
child41.text = '458'

child42 = etree.SubElement(child4,'height')
child42.text = '45'

child43 = etree.SubElement(child4,'height')
child43.text = '45'

child5 = etree.SubElement(root,'segmented')
child5.text = '0'
#自定义数据集
objectlist = [{'xmin':263,'ymin':211,'xmax':324,'ymax':339},
              {'xmin':5,'ymin':224,'xmax':67,'ymax':374}]


for i in objectlist:
    
    child6 = etree.SubElement(root,'object')

    child61 = etree.SubElement(child6,'name')
    child61.text = 'face'

    child62 = etree.SubElement(child6,'pose')
    child62.text = 'Unspecified'

    child63 = etree.SubElement(child6,'truncated')
    child63.text = '0'

    child64 = etree.SubElement(child6,'difficult')
    child64.text = '0'

    child65 = etree.SubElement(child6,'bndbox')

    child651 = etree.SubElement(child65,'xmin')
    child651.text = str(i['xmin'])

    child652 = etree.SubElement(child65,'ymin')
    child652.text = str(i['ymax'])

    child653 = etree.SubElement(child65,'xmax')
    child653.text = str(i['xmax'])

    child654 = etree.SubElement(child65,'ymax')
    child654.text = str(i['ymax'])

tree = etree.ElementTree(root)  
tree.write('654.xml', pretty_print=True, xml_declaration=True, encoding='utf-8')

参考https://blog.csdn.net/qq_16209077/article/details/55049877