|
python解析xml文件,目标检测的voc格式的数据
- # 读取文件
- # xml_annotation_path = "*.xml" from xml.dom.minidom import parse
- dom = parse(xml_annotation_path)
- # 获取文档元素对象
- data = dom.documentElement
- objs = data.getElementsByTagName('object')
- for obj in objs:
- # 获取标签中内容
- name = obj.getElementsByTagName('name')[0].childNodes[0].nodeValue
- x1 = obj.getElementsByTagName('bndbox')[0].getElementsByTagName('xmin')[0].childNodes[0].nodeValue
- y1 = obj.getElementsByTagName('bndbox')[0].getElementsByTagName('ymin')[0].childNodes[0].nodeValue
- x2 = obj.getElementsByTagName('bndbox')[0].getElementsByTagName('xmax')[0].childNodes[0].nodeValue
- y2 = obj.getElementsByTagName('bndbox')[0].getElementsByTagName('ymax')[0].childNodes[0].nodeValue
- print('name:', name, ', box:[{},{},{},{}]'.format(x1, y1, x2, y2))
- assert 0 == 1, "停"
复制代码
|
|