python读取xml数据

XML解析与目标检测标注

最新推荐文章于 2024-08-02 05:47:03 发布

原创最新推荐文章于 2024-08-02 05:47:03 发布 · 752 阅读

2 ·

CC 4.0 BY-SA版权

文章标签：

#python #xml

python 专栏收录该内容

2 篇文章

订阅专栏

本文介绍了一种从XML文件中解析目标检测标注数据的方法。通过Python的xml.dom.minidom模块读取XML文件，获取图像中物体的类别名称及边界框坐标。此过程为后续的目标检测模型训练提供数据准备。

也是直接给出一个小栗子

from xml.dom import minidom

FILE_PATH = 'Annotation/1.xml'
try:
    doc = minidom.parse(FILE_PATH)
except Exception, e:
    print FILE_PATH + ' wrong.'
    print e

root = doc.documentElement
objname = root.getElementsByTagName('name')[0].firstChild.data
cls = imdbcls.imdb_cls_name(str(objname).lower().strip())
    
gtnum = root.getElementsByTagName('name').length

gt_boxes = []
for lyi in range(gtnum):
    gtbox = []
    gtbox.append(int(root.getElementsByTagName("xmin")[lyi].firstChild.data))
    gtbox.append(int(root.getElementsByTagName("ymin")[lyi].firstChild.data))
    gtbox.append(int(root.getElementsByTagName("xmax")[lyi].firstChild.data))
    gtbox.append(int(root.getElementsByTagName("ymax")[lyi].firstChild.data))
    gt_boxes.append(gtbox)