东方耀AI技术分享

标题: python解析xml文件,目标检测的voc格式的数据 [打印本页]

作者: 东方耀    时间: 2020-11-3 17:23
标题: python解析xml文件,目标检测的voc格式的数据


python解析xml文件,目标检测的voc格式的数据

  1. # 读取文件
  2.     # xml_annotation_path = "*.xml"  from xml.dom.minidom import parse
  3.     dom = parse(xml_annotation_path)
  4.     # 获取文档元素对象
  5.     data = dom.documentElement
  6.     objs = data.getElementsByTagName('object')
  7.     for obj in objs:
  8.         # 获取标签中内容
  9.         name = obj.getElementsByTagName('name')[0].childNodes[0].nodeValue
  10.         x1 = obj.getElementsByTagName('bndbox')[0].getElementsByTagName('xmin')[0].childNodes[0].nodeValue
  11.         y1 = obj.getElementsByTagName('bndbox')[0].getElementsByTagName('ymin')[0].childNodes[0].nodeValue
  12.         x2 = obj.getElementsByTagName('bndbox')[0].getElementsByTagName('xmax')[0].childNodes[0].nodeValue
  13.         y2 = obj.getElementsByTagName('bndbox')[0].getElementsByTagName('ymax')[0].childNodes[0].nodeValue
  14.         print('name:', name, ', box:[{},{},{},{}]'.format(x1, y1, x2, y2))
  15.     assert 0 == 1, "停"
复制代码



作者: bixintao    时间: 2020-11-10 19:50
积分积分积分




欢迎光临 东方耀AI技术分享 (http://www.ai111.vip/) Powered by Discuz! X3.4