Python基础知识之读取XML文件
admin
2024-04-29 19:32:15
0次

        如果你是深度学习的一员,那么平时在训练模型的时候,就会常看到一些标注是使用xml文件来做的,比如常用到的一个数据集VOC,它的Annotations就是用xml来储存图片中对象的属性的。
所以对于熟悉读取xml文件是很有必要的,我们先来看个xml文件,来自VOC2012的对2007_000027.jpg图片里面对象属性的标注,2007_000027.xml:

VOC20122007_000027.jpgThe VOC2007 DatabasePASCAL VOC2007flickr48650030personUnspecified00174101349351head169104209146hand278210297233foot273333297354foot319307340326

那如何来读取里面的节点与它的内容呢,这里需要使用到xml.etree.ElementTree这样的一个库

import xml.etree.ElementTree as ET
tree=ET.parse('2007_000027.xml')

通过parse解析文件,生成一个树的对象,然后在这个对象里面去寻找节点即可
比如这里,我们想要读取图片的宽和高节点的属性值,也就是节点size里面的width与height,通过tree中的find来查找:

tree.find('size').find('width').text
tree.find('size').find('height').text

注意的是,这里查找出来的486和500的值,是str类型而不是int,所以如果是需要为整型类型时,需要做类型转换

如果是多个同样节点,比如多个节点,如何读取呢?
这里使用findall来进行遍历访问,返回的是列表,也就是find获取的元素的集合,我们来看下多个节点的xml文件2007_000032.xml:

VOC20122007_000032.jpgThe VOC2007 DatabasePASCAL VOC2007flickr50028131aeroplaneFrontal0010478375183aeroplaneLeft0013388197123personRear00195180213229personRear002618944238

可以看到有四个分别存储着四个不同的对象,我们来遍历这个节点,顺便将里面的锚框节点的四个属性(左上角和右下角的坐标)也给查找出来,这里我们通过字典保存每个对象方便后期调用,然后将字典都添加到一个列表里面:

tree = ET.parse('2007_000032.xml')
objects = []
for obj in tree.findall('object'):obj_dict = dict()obj_dict['name'] = obj.find('name').textbbox = obj.find('bndbox')obj_dict['bbox'] = [int(float(bbox.find('xmin').text)), int(float(bbox.find('ymin').text)), int(float(bbox.find('xmax').text)), int(float(bbox.find('ymax').text))]objects.append(obj_dict)
print(len(objects),objects)
print(objects[3]['bbox'])'''
4 [{'name': 'aeroplane', 'bbox': [104, 78, 375, 183]}, {'name': 'aeroplane', 'bbox': [133, 88, 197, 123]}, {'name': 'person', 'bbox': [195, 180, 213, 229]}, {'name': 'person', 'bbox': [26, 189, 44, 238]}]
[26, 189, 44, 238]
'''

相关内容

热门资讯

AI做攻略、“双向奔赴”……国... 本文转自:人民网人民网记者 许维娜今年国庆假期,AI(人工智能)成了不少年轻人的“旅行搭子”。行前做...
合肥的这个“梗”, 翻篇了 社交平台的评论区,曾长期流传一句专属于合肥的黑色幽默。有人问合肥哪里好玩,本地网友半开玩笑地回复:合...
7万+旅游中高层都在的群,今天... 做旅游这行,资源卡位、渠道互通、项目落地——往往就差一个“对的机会”。我们的行业社群已聚集70,00...
一场帐篷音乐节,如何让井研从“... 帐篷音乐节现场封面新闻记者 周洪攀10月的研溪湿地,晚风里裹着柑橘的清甜与民谣的回响。夜幕降临,10...
白鹤翩翩落星愿,上海迪士尼度假... 10月9日,世界候鸟日前夕,上海迪士尼度假区星愿公园的湖风里多了一分白鹤的灵动。三座白鹤雕塑正式揭幕...
成都锦江边赏初秋枫叶飘飘 目前,成都锦江边国华街枫叶树红了。近年来,国华街附近的几条街都栽种了红枫树,是城区较集中看枫叶的点位...
GBTA:商务旅行正在复苏,但... 亚太正在成为全球商务旅行增长的重要引擎。在经历疫情后的快速反弹之后,全球商务旅行正在进入一个新的增长...
从景区焕新到节赛引流 邹城双节... 齐鲁网·闪电新闻10月9日讯 中秋、国庆“双节”假期,邹城市文化和旅游局提前谋划、周密部署,立足自身...
江山如此多娇 且看石楼今朝 金秋十月,黄河岸畔秋意盎然,丰收盛会喜迎国庆。本次石楼国庆文旅活动以“江山如此多娇,且看石楼今朝”为...
海南国庆假期接待游客475.4... 2026年国庆假期,海南接待游客475.40万人次,较2025年增长13.1%,较2024年增长15...