极简Python学习-----XML文件解析

最新推荐文章于 2024-08-06 17:00:20 发布

King的王国

最新推荐文章于 2024-08-06 17:00:20 发布

阅读量257

点赞数

分类专栏： Python 文章标签： python

本文链接：https://blog.csdn.net/qq_28057379/article/details/105674939

版权

Python 专栏收录该内容

34 篇文章 2 订阅

订阅专栏

以下代码为：采用labelImage工具标注数据后，通过解析xml文件，得到标注框。

import xml.etree.ElementTree as ET
import glob
import cv2

xmls_dir = glob.glob('C:/Users/97875/Desktop/data_test_20200421/data_test_20200421_0000.xml')
ii = 0
for xml_dir in xmls_dir:
     #加载图片
     img_dir = xml_dir.replace('xml', 'jpg')
     img = cv2.imread(img_dir)

     #返回一个ElementTree实例
     tree = ET.parse(xml_dir)
     #返回此树的根元素
     root = tree.getroot()
     print(root)
     #root.tag：获取节点的名称， root.attrib：节点的属性列表， root.text：节点的内容
     print('root-tag:',root.tag,',root-attrib:',root.attrib,',root-text:',root.text)
     #按标记名或路径查找所有匹配的子元素。
     for item in root.iterfind('object'):
          print('item', item)
          for i in item.iterfind('bndbox'):
               img_roi = img[int(i[1].text):int(i[3].text), int(i[0].text):int(i[2].text)]
               cv2.imwrite('./roi1/' + str(ii) + '.jpg', img_roi)
               ii += 1