[python]利用Python生成xml文件

最新推荐文章于 2023-09-02 14:10:26 发布
最新推荐文章于 2023-09-02 14:10:26 发布
阅读量4.3k 收藏 8
点赞数 1
分类专栏： python
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/u013250416/article/details/78429281
版权

写在前面：

github上的Text-Detection-with-FRCN代码是py-faster-cnn在场景文字识别上面的扩展，基本只是把数据集和分类类别进行了改变。在样例中，将原有的pascal_voc数据集换成了coco_text。而如果想要在代码中不进行大的改动，最好的方式就是统一两个数据集的格式。

在github中，提供了format data的方式：

# format the raw image and label into the type of pascal_voc

# follow the code in $Text-Detection-with-FRCN/datasets/script/format_annotation.p

cd$Text-Detection-with-FRCN/datasets/script

./format_annotation.py --dataset coco-text

我们来看一下format_annotation.py的倒数第二行， os.system('./ann2voc2007.sh'+ args.dataset)

‘./ann2voc2007.sh'的内容为：matlab -nodisplay -nodesktop -r"ann2voc2007('$1'); quit"　

也就是用matlib来运行。下载matlib要花费的时间很长，如果不想安装，怎么办呢？可以利用python来对matlib文件进行改写。

matlib的内容是生成xml文件。那么相应的，也可以利用python生成xml文件。

原matlib代码：

function ann2voc2007(input_dir)
  curpath = mfilename('fullpath');
  [pathstr,~,~] = fileparts(curpath)
  if input_dir(end) == '/'
    input_dir = input_dir(1:end-1);
  end
  [~,input_dir,~] = fileparts(input_dir);
  input_dir = [pathstr '/../' input_dir '/formatted_dataset']
  imgpath = [input_dir '/JPEGImages/']
  txtpath = [input_dir '/images.annotations']
  xmlpath_new = [input_dir '/Annotations/'];
  foldername = 'VOC2007';
  coco = containers.Map();
  fidin = fopen(txtpath, 'r');
  cnt = 0;
  while ~feof(fidin)
    tline = fgetl(fidin);
    str = regexp(tline, ' ', 'split');
    xmlname = strrep(str{1},'.jpg','.xml');
    info = imfinfo([imgpath '/' str{1}]);
    str{3} = max(str2double(str{3}), 1);
    str{4} = max(str2double(str{4}), 1);
    str{5} = min(str2double(str{5}), info.Width);
    str{6} = min(str2double(str{6}), info.Height);
    if str{3} >= str{5} || str{4} >= str{6} || str{3} <= 0 || str{4} <= 0 || str{5} > info.Width...
      str{6} > info.Height
      continue;
    end
    cnt = cnt + 1
    if exist([imgpath '/' str{1}])
      if isKey(coco,xmlname)
        Createnode = coco(xmlname);
        object_node = Createnode.createElement('object');
        Root = Createnode.getDocumentElement;
        Root.appendChild(object_node);
        node=Createnode.createElement('name');
        node.appendChild(Createnode.createTextNode(str{2}));
        object_node.appendChild(node);
        node=Createnode.createElement('pose');
        node.appendChild(Createnode.createTextNode('Unspecified'));
        object_node.appendChild(node);
        node=Createnode.createElement('truncated');
        node.appendChild(Createnode.createTextNode('0'));
        object_node.appendChild(node);
        node=Createnode.createElement('difficult');
        node.appendChild(Createnode.createTextNode('0'));
        object_node.appendChild(node);
        bndbox_node=Createnode.createElement('bndbox');
        object_node.appendChild(bndbox_node);
        node=Createnode.createElement('xmin');
        node.appendChild(Createnode.createTextNode(num2str(str{3})));
        bndbox_node.appendChild(node);
        node=Createnode.createElement('ymin');
        node.appendChild(Createnode.createTextNode(num2str(str{4})));
        bndbox_node.appendChild(node);
        node=Createnode.createElement('xmax');
        node.appendChild(Createnode.createTextNode(num2str(str{5})));
        bndbox_node.appendChild(node);
        node=Createnode.createElement('ymax');
        node.appendChild(Createnode.createTextNode(num2str(str{6})));
        bndbox_node.appendChild(node);
      else
        Createnode = com.mathworks.xml.XMLUtils.createDocument('annotation');
        Root = Createnode.getDocumentElement;
        node = Createnode.createElement('folder');
        node.appendChild(Createnode.createTextNode(foldername));
        Root.appendChild(node);
        node = Createnode.createElement('filename');
        node.appendChild(Createnode.createTextNode(str{1}));
        Root.appendChild(node);
        source_node = Createnode.createElement('source');
        Root.appendChild(source_node);
        node = Createnode.createElement('database');
        node.appendChild(Createnode.createTextNode('MS COCO-Text'));
        source_node.appendChild(node);
        node = Createnode.createElement('annotation');
        node.appendChild(Createnode.createTextNode('MS COCO-Text 2014'));
        source_node.appendChild(node);
        node=Createnode.createElement('image');
        node.appendChild(Createnode.createTextNode('NULL'));
        source_node.appendChild(node);
        node=Createnode.createElement('flickrid');
        node.appendChild(Createnode.createTextNode('NULL'));
        source_node.appendChild(node);
        owner_node=Createnode.createElement('owner');
        Root.appendChild(owner_node);
        node=Createnode.createElement('flickrid');
        node.appendChild(Createnode.createTextNode('NULL'));
        owner_node.appendChild(node);
        node=Createnode.createElement('name');
        node.appendChild(Createnode.createTextNode('ligen'));
        owner_node.appendChild(node);
        size_node=Createnode.createElement('size');
        Root.appendChild(size_node);
        node=Createnode.createElement('width');
        node.appendChild(Createnode.createTextNode(num2str(info.Width)));
        size_node.appendChild(node);
        node=Createnode.createElement('height');
        node.appendChild(Createnode.createTextNode(num2str(info.Height)));
        size_node.appendChild(node);
        node=Createnode.createElement('depth');
        node.appendChild(Createnode.createTextNode(num2str(info.BitDepth / 8)));
        size_node.appendChild(node);
        node=Createnode.createElement('segmented');
        node.appendChild(Createnode.createTextNode('0'));
        Root.appendChild(node);
        object_node=Createnode.createElement('object');
        Root.appendChild(object_node);
        node=Createnode.createElement('name');
        node.appendChild(Createnode.createTextNode(str{2}));
        object_node.appendChild(node);
        node=Createnode.createElement('pose');
        node.appendChild(Createnode.createTextNode('Unspecified'));
        object_node.appendChild(node);
        node=Createnode.createElement('truncated');
        node.appendChild(Createnode.createTextNode('0'));
        object_node.appendChild(node);
        node=Createnode.createElement('difficult');
        node.appendChild(Createnode.createTextNode('0'));
        object_node.appendChild(node);
        bndbox_node=Createnode.createElement('bndbox');
        object_node.appendChild(bndbox_node);
        node=Createnode.createElement('xmin');
        node.appendChild(Createnode.createTextNode(num2str(str{3})));
        bndbox_node.appendChild(node);
        node=Createnode.createElement('ymin');
        node.appendChild(Createnode.createTextNode(num2str(str{4})));
        bndbox_node.appendChild(node);
        node=Createnode.createElement('xmax');
        node.appendChild(Createnode.createTextNode(num2str(str{5})));
        bndbox_node.appendChild(node);
        node=Createnode.createElement('ymax');
        node.appendChild(Createnode.createTextNode(num2str(str{6})));
        bndbox_node.appendChild(node);
        coco(xmlname) = Createnode;
      end
    end
  end
  fclose(fidin);
  keyss = keys(coco);
  for i = 1:length(keyss)
    xmlwrite([xmlpath_new '/' keyss{i}], coco(keyss{i}));
  end
end


改写后的python代码： 

#coding:utf-8
from PIL import Image
from xml.dom.minidom import Document
import os

def main():
	imgpath = 'JPEGImages/'
	txtpath = 'images.annotations'
	xmlpath_new = 'Annotations/'
	coco = {}
	
	#　得到图像的标注信息
	file_object = open(txtpath,'rU')
	try: 
	    for line in file_object:
			line = line.rstrip('\n')
			strs = line.split(' ')
			print strs[0]
			foldername = 'VOC2007'
			
			#　用xml替换jpg，得到同名文件
			xmlname = strs[0].replace('.jpg','.xml')
			info = Image.open(imgpath + strs[0])
			# read image size
			(width,height) = info.size
			strs[2] = max(float(strs[2]), 1)
			strs[3] = max(float(strs[3]), 1)
			strs[4] = min(float(strs[4]), width);
			strs[5] = min(float(strs[5]), height);

			# 过滤异常
			if strs[2] >= strs[4] or strs[3] >= strs[5] or strs[2] <=0 or strs[3] <= 0 or strs[4] > width or strs[5] > height:
				continue
			
			if os.path.exists(imgpath + strs[0]):
				if xmlname in coco:
					Createnode = coco[xmlname]
					object_node = Createnode.createElement('object')
					
					Root = Createnode.getElementsByTagName('annotation')[0]
					Root.appendChild(object_node)
					
					node=Createnode.createElement('name')
					node.appendChild(Createnode.createTextNode(strs[1]))
					object_node.appendChild(node)
					
					node=Createnode.createElement('pose')
					node.appendChild(Createnode.createTextNode('Unspecified'))
					object_node.appendChild(node)
					
					node=Createnode.createElement('truncated')
					node.appendChild(Createnode.createTextNode('0'))
					object_node.appendChild(node)
					
					node=Createnode.createElement('difficult')
					node.appendChild(Createnode.createTextNode('0'))
					object_node.appendChild(node)
					
					bndbox_node=Createnode.createElement('bndbox')
					object_node.appendChild(bndbox_node)
					
					node=Createnode.createElement('xmin')
					node.appendChild(Createnode.createTextNode(str(strs[2])))
					bndbox_node.appendChild(node)
					
					node=Createnode.createElement('ymin')
					node.appendChild(Createnode.createTextNode(str(strs[3])))
					bndbox_node.appendChild(node)
					
					node=Createnode.createElement('xmax')
					node.appendChild(Createnode.createTextNode(str(strs[4])))
					bndbox_node.appendChild(node)
					
					node=Createnode.createElement('ymax')
					node.appendChild(Createnode.createTextNode(str(strs[5])))
					bndbox_node.appendChild(node)
				else:
					Createnode=Document()  #创建DOM文档对象
					
					Root=Createnode.createElement('annotation') #创建根元素
					Createnode.appendChild(Root)
					
					# folder
					folder=Createnode.createElement('folder')
					folder.appendChild(Createnode.createTextNode(foldername))
					Root.appendChild(folder)
					
					# filename
					filename = Createnode.createElement('filename')
					filename.appendChild(Createnode.createTextNode(strs[0]))
					Root.appendChild(filename)
					
					# source
					source_node = Createnode.createElement('source')
					Root.appendChild(source_node)
					
					node = Createnode.createElement('database')
					node.appendChild(Createnode.createTextNode('MS COCO-Text'))
					source_node.appendChild(node)
					
					node = Createnode.createElement('annotation')
					node.appendChild(Createnode.createTextNode('MS COCO-Text 2014'))
					source_node.appendChild(node)
			
					node=Createnode.createElement('image')
					node.appendChild(Createnode.createTextNode('NULL'))
					source_node.appendChild(node)
			
					node=Createnode.createElement('flickrid');
					node.appendChild(Createnode.createTextNode('NULL'));
					source_node.appendChild(node);
					
					# owner
					owner_node=Createnode.createElement('owner')
					Root.appendChild(owner_node)
					
					node=Createnode.createElement('flickrid')
					node.appendChild(Createnode.createTextNode('NULL'))
					owner_node.appendChild(node)
					
					node=Createnode.createElement('name')
					node.appendChild(Createnode.createTextNode('ligen'))
					owner_node.appendChild(node)
					
					# size
					size_node=Createnode.createElement('size')
					Root.appendChild(size_node)
					
					node=Createnode.createElement('width')
					node.appendChild(Createnode.createTextNode(str(width)))
					size_node.appendChild(node)
					
					node=Createnode.createElement('height');
					node.appendChild(Createnode.createTextNode(str(height)))
					size_node.appendChild(node)
					
					node=Createnode.createElement('depth')
					node.appendChild(Createnode.createTextNode('3'))
					size_node.appendChild(node)
					
					# segmented
					node=Createnode.createElement('segmented')
					node.appendChild(Createnode.createTextNode('0'))
					Root.appendChild(node)
					
					# object
					object_node=Createnode.createElement('object')
					Root.appendChild(object_node)
					
					node=Createnode.createElement('name')
					node.appendChild(Createnode.createTextNode(strs[1]))
					object_node.appendChild(node)
					
					node=Createnode.createElement('pose')
					node.appendChild(Createnode.createTextNode('Unspecified'))
					object_node.appendChild(node)
					
					node=Createnode.createElement('truncated')
					node.appendChild(Createnode.createTextNode('0'))
					object_node.appendChild(node)
					
					node=Createnode.createElement('difficult')
					node.appendChild(Createnode.createTextNode('0'))
					object_node.appendChild(node)
					
					bndbox_node=Createnode.createElement('bndbox')
					object_node.appendChild(bndbox_node)
					
					node=Createnode.createElement('xmin')
					node.appendChild(Createnode.createTextNode(str(strs[2])))
					bndbox_node.appendChild(node)
					
					node=Createnode.createElement('ymin')
					node.appendChild(Createnode.createTextNode(str(strs[3])))
					bndbox_node.appendChild(node)
					
					node=Createnode.createElement('xmax')
					node.appendChild(Createnode.createTextNode(str(strs[4])))
					bndbox_node.appendChild(node)
					
					node=Createnode.createElement('ymax')
					node.appendChild(Createnode.createTextNode(str(strs[5])))
					bndbox_node.appendChild(node)
					
					coco[xmlname] = Createnode
			
	finally:
	     file_object.close()
	print 'begin load xml...'
	for key in coco:
		print key
		f = open(xmlpath_new + key,'w')
		f.write(coco[key].toprettyxml(indent = '\t'))
		f.close()

if __name__ == "__main__":
	main()


可以看出，在生成ｘｍｌ文件的方式上面，二者很相像。

(1)matlib创建文档节点及根元素的方式：

        Createnode = com.mathworks.xml.XMLUtils.createDocument('annotation');
        Root = Createnode.getDocumentElement;
        node = Createnode.createElement('folder');
        node.appendChild(Createnode.createTextNode(foldername));
        Root.appendChild(node);

python创建文档节点及根元素的方式：

					Createnode=Document()  #创建DOM文档对象
					
					Root=Createnode.createElement('annotation') #创建根元素
					Createnode.appendChild(Root)
					
					# folder
					folder=Createnode.createElement('folder')
					folder.appendChild(Createnode.createTextNode(foldername))
					Root.appendChild(folder)

(2)matlib获取文档已存在的根元素的方式：

        Createnode = coco(xmlname);
        object_node = Createnode.createElement('object');
        Root = Createnode.getDocumentElement;
        Root.appendChild(object_node);

python获取文档已存在的根元素的方式： 

					Createnode = coco[xmlname]
					object_node = Createnode.createElement('object')
					
					Root = Createnode.getElementsByTagName('annotation')[0]
					Root.appendChild(object_node)



matlib转python还需要注意的地方：

１）matlib中list的索引是从１开始，而python中是从０开始

２）python中str是关键字


总结：

其实python转xml并不是什么新东西。写这篇博文的目的，在于纪念这次思考问题的方式。

在实验环境不能够满足要求时，例如本次缺少matlib，考虑时间成本，是为了这部分代码下载matlib还是重写呢？衡量二者的复杂度和时间成本，选择了重写。




crediks
关注
  • 1
    点赞
  • 踩
  • 8
    收藏
    觉得还不错? 一键收藏
  • 2
    评论
专栏目录
python挑出训练集里图片对应的xml文件，方便统计标签框的类别与数目_python 统计voc2007xml中某一类别框个数
m0_60635176的博客
04-15 883
书籍的好处就在于权威和体系健全，刚开始学习的时候你可以只看视频或者听某个人讲课，但等你学完之后，你觉得你掌握了，这时候建议还是得去看一下书籍，看权威技术书籍也是每个程序员必经之路。Python所有方向路线就是把Python常用的技术点做整理，形成各个领域的知识点汇总，它的用处就在于，你可以按照上面的知识点去找对应的学习资源，保证自己学得较为全面。‘’‘source_dir=’/home/…/VOC2007/ImageSets/Main/train’#在该文件得到文件名。工欲善其必先利其器。
【深度学习】检测结果转成XML
wangxiaobei2017的博客
12-15 1176
检测结果转成XML检测图像标注现有xml增加检测框数据后的xml具体实现 检测图像标注 在图像标注过程中，为了减少大量标注工作，可以预先训练小模型，然后用该模型对未标注图像进行推理，将检测框角点数据转成XML。 再利用labelimg对未标注图像进行加载，同时加载生成的XML，可以减少一部分数据标注工作 现有xml <?xml version="1.0" ?><annotation> <folder>fold</folder> <filename
Python如何生成xml文件
09-16
在本篇文章里小编给大家分享的是一篇关于Python生成xml文件方法以及实例代码，需要的朋友们可以学习下。
使用python批量修改图像尺寸、文件名称、xml文件内容、生成无标记xml文件！
最新发布
一米阳光-的博客
09-02 327
深度学习中经常需要训练无标记的负样本，用于降低目标检测中的误检率、误识别率。生成负样本xml文件的一个方法是用labelImg在图像上随便画一个框，生成xml文件，然后手动去除相关的object节点，但是负样本太多的话，这种方法太耗时间，所以我写了一个脚本可以批量生成空的xml文件。此外本博客还包含了如何批量修改图像尺寸、文件名称、xml文件内容，这些也经常用到，代码如下，大家可以适当参考。小编水平有限，如有错误，麻烦各位告诉我一下哈！
深度学习之数据增强脚本（支持Cutout，旋转，平移等操作并自动生成新的xml文件）
m0_59208577的博客
04-12 2085
在学习目标检测时，有时会发现数据量不够，这时候便需要对数据进行扩充，这里给出一个数据增强脚本，可用于使用目标检测算法时数据量太少的情况。 该脚本支持对输入的图片进行Cutout，旋转、翻转、平移、加噪声、调整亮度、剪裁、镜像等操作，输入图片文件夹的路径、以及对应xml文件的路径，运行代码即可进行数据增强，新生成的图片以及xml文件会保存在原图片以及xml路径下，对图片进行随机的操作后，会对xml文件的内容进行相应的修改，不需要自己手动重新标注。
UTKinect-Action数据库中xml数据读取成mat文件
03-16
matlab环境下，将UTKinect-Action数据库中xml数据读取成mat文件
Python根据指定文件生成XML的方法
09-16
主要介绍了Python根据指定文件生成XML的方法，文中代码非常详细，帮助大家更好的理解和学习，感兴趣的朋友可以了解下
利用 Python ElementTree 生成 xml的实例
09-17
在这个实例中，我们将详细探讨如何使用`ElementTree`来生成XML文件。 首先，我们需要导入`ElementTree`模块，通常我们会将其别名为`etree`，以便后续代码更加简洁： ```python from xml.etree import ElementTree ...
Python中使用dom模块生成XML文件示例
09-22
主要介绍了Python中使用dom模块生成XML文件示例,本文讲解了dom模块的主要方法、详细的代码实例,需要的朋友可以参考下
XML4MAT v2.0：将 XML 与 MATLAB 结构化变量相互转换。-matlab开发
06-01
MATLAB 结构化变量和 XML 字符串之间的相互转换。 定义了一种标记语言以支持任何 MATLAB 变量的字符串表示。 附加的 m 函数支持任何 XML 结构的导入。 最后一张图：任何 XML --> MbML <--> MATLAB。 描述 tbox 原版 (v1.0) 的手稿发表于： Almeida, JS、Wu, S. 和 Voit, EO (2003) XML4MAT：MatlabTM 结构化变量和标记语言 MbML 之间的相互转换。 计算机科学预印本档案（Elsevier），2003（12）：9-17。
Python创建xml文件示例
09-21
主要介绍了Python创建xml文件的方法,结合实例形式分析了Python针对xml格式数据及文件读写相关操作技巧,需要的朋友可以参考下
怎么用python创建文件-如何用Python创建生成xml文档文件的方法
weixin_37988176的博客
11-01 233
用如何用Python完成创建生成xml文档文件的工作。需要用到的方法：from xml.dom.minidom import Document，想必大家很熟悉这个操作了，接下来就用python 创建create xml。python目录里面创建xml包，好了下面就开始我们正式的创建xml文档文件的python代码部分了：from xml.dom.minidom import Document#导入...
Object Recognition and Scene Understanding（九）模型转化Mat-〉XML
loadstar_kun的专栏
04-03 2761
第一步：获取参数 其中为matlab语言 1.components 个数 NumComponents= length(model.rules{model.start}) ncom =length(model.rules{model.start});  2.Numberof features这个固定， = 31，model文件中没有记录  3.Score threshold= model
shp转geojson（python）
debugcool的博客
03-05 1981
shp转为geojson 首先安装pyshp pip install pyshp import shapefile from json import dumps reader = shapefile.Reader(&quot;E:\Work\data\shp\example.shp&quot;) fields = reader.fields[1:] field_names = [field[0] for fiel...
python 输出xml格式文件
u013862444的博客
07-12 1974
File "C:\ProgramData\Anaconda3\lib\xml\etree\ElementTree.py", line 1058, in _raise_serialization_error "cannot serialize %r (type %s)" % (text, type(text).__name__) TypeError: cannot serialize 1 (...
python创建xml文档
haimianjie2012的专栏
12-06 1232
Python minidom模块(DOM写入和解析XML) 用python快速制作xml文件 python生成xml文件
python格式化输出xml_如何用Python输出漂亮的xml文件
weixin_39789499的博客
12-12 1723
匿名用户1级2017-07-25 回答最近要用Python处理一个xml文件。平时习惯了用xml.etree.ElementTree，结果发现此库存在一些让人无法忍受的缺点：1. 无法处理comment和cdata2. 输出后的xml文件没有换行，没有缩进，十分难以阅读于是尝试这用xml.dom.minidom，前两个问题几乎全部解决了。唯一让人看着不爽的是minidom对于text的处理。比如，...
python 在指定路径生成xml文件
04-18
好的，你可以使用Python中的xml.etree.ElementTree模块来生成XML文件，首先需要导入该模块，并创建根元素。然后，你可以添加子元素和它们的属性和文本。最后，使用ElementTree的write()方法将XML输出到指定文件路径中。下面是一个简单的示例代码： ```python import xml.etree.ElementTree as ET # 创建根元素 root = ET.Element("persons") # 添加子元素 person1 = ET.SubElement(root, "person") person1.set("id", "001") name1 = ET.SubElement(person1, "name") name1.text = "John" age1 = ET.SubElement(person1, "age") age1.text = "30" person2 = ET.SubElement(root, "person") person2.set("id", "002") name2 = ET.SubElement(person2, "name") name2.text = "Alice" age2 = ET.SubElement(person2, "age") age2.text = "25" # 写入xml文件 tree = ET.ElementTree(root) tree.write("path/to/xml/file.xml") ``` 请注意，path/to/xml/file.xml应替换为您希望生成XML文件的实际路径。

“相关推荐”对你有帮助么？

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 2
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值