lvsling

  最近生成训练数据时,给一批无效的背景图片生成对应的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()

  

分类:

技术点:

相关文章: