将目标检测的xml标签中目标绘制在图像中

用xml.etree.ElementTree库来解析xml文件

1.解析:ET.parse()

2.获取根节点:getroot()

3.找节点用find() 和findall()

4.

   .text 是内容 #本代码用到.text

   .tag 是题目

   .attrib 是属性

5.用cv2绘图

提取出来坐标,然后rectangle(图片,左上角坐标,右下角坐标,框颜色,框线粗细)

把目标名字输出:putTEXT(图片,目标名字,起始坐标,字体,字体大小,字体颜色,粗细)

其他:两个find()定位到节点下,才能去提取xmin等四个值。还有另一个xml解析库,为xml.dom.minidom

没用过,但看起来差不多。效果都一样,熟悉一个就行。

# -*- coding: utf-8 -*-
import xml.etree.ElementTree as ET  
import os,cv2
 
xml_file='C:/Users/nansbas/Desktop/01_000002_01244-01086_0939-0989.xml'
tree=ET.parse(xml_file)
root=tree.getroot()
imgfile='C:/Users/nansbas/Desktop/01_000002_01244-01086_0939-0989.jpg'
im = cv2.imread(imgfile)
for object in root.findall('object'):
    object_name

你可能感兴趣的:(数据清洗)