Minimal Python learning-----XML file parsing

以下代码为:采用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

Guess you like

Origin blog.csdn.net/qq_28057379/article/details/105674939