用python批量生成简单的xml文档

时间：2023-03-09 03:32:16
用python批量生成简单的xml文档

　　最近生成训练数据时，给一批无效的背景图片生成对应的xml文档，我用python写了一个简单的批量生成xml文档的demo，遇见了意外的小问题，记录一下。

　　报错问题为：ImportError: No module named 'xml.dom'; 'xml' is not a package

　　看见No module named “xxx”时想的是不就是没安装xml包嘛，还不简单，install一下不就好了，然而并没什么用，xml是python本生就带的。其实出现这种错误的原因是自己的命名规则问题，将python程序命名为"xml.py"导致的。让运行python程序时出现自导现象，从而出错。一个小错误，发现时我自己都笑了。

　　顺便贴一下代码，作为一个小demo分享一下。

　　

import os

import xml.dom.minidom

read_file='jpg'

for file_name in os.listdir(read_file):

    new_txtname=file_name.split('.')[0]

    #创建一个空的Dom文档对象

    doc = xml.dom.minidom.Document()

    #创建根节点，此根节点为annotation

    annotation = doc.createElement('annotation')

    #将根节点添加到DOm文档对象中

    doc.appendChild(annotation) 

    folder = doc.createElement('folder')

    #内容写入

    folder_text = doc.createTextNode('ee')

    folder.appendChild(folder_text)

    annotation.appendChild(folder)

    filename = doc.createElement('filename')

    filename_text = doc.createTextNode(file_name)

    filename.appendChild(filename_text)

    annotation.appendChild(filename)

    path = doc.createElement('path')

    path_text = doc.createTextNode('path is null')

    path.appendChild(path_text)

    annotation.appendChild(path)

    source = doc.createElement('source')

    databass = doc.createElement('databass')

    databass_text = doc.createTextNode('Unknown')

    source.appendChild(databass)

    databass.appendChild(databass_text)

    annotation.appendChild(source)

    size = doc.createElement('size')

    width = doc.createElement('width')

    width_text = doc.createTextNode('875')

    height = doc.createElement('height')

    height_text = doc.createTextNode('656')

    depth = doc.createElement('depth')

    depth_text = doc.createTextNode('1')

    size.appendChild(width)

    width.appendChild(width_text)

    size.appendChild(height)

    height.appendChild(height_text)

    size.appendChild(depth)

    depth.appendChild(depth_text)

    annotation.appendChild(size)

    segmented = doc.createElement('segmented')

    segmented_text = doc.createTextNode('0')

    segmented.appendChild(segmented_text)

    annotation.appendChild(segmented)

    #写入xml文本文件中

    fp = open('xml/%s.xml' %new_txtname , 'w+')

    doc.writexml(fp, indent='\t', addindent='\t', newl='\n',encoding='utf-8')

    fp.close()

　　

相关文章