Python显示Ground True

最新推荐文章于 2024-11-25 15:15:43 发布

转载最新推荐文章于 2024-11-25 15:15:43 发布 · 384 阅读

0 ·

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/vshen999/p/11270550.html

文章标签：

#python #matlab

本文介绍了一种使用Python读取PascalVOC2007数据集中图像及其XML标注文件的方法，并展示了如何从XML文件中提取目标框坐标，以及如何在图像上绘制这些边界框。

数据使用Pascal voc 2007 里边的000012图像，将这张图片和对应的xml文件复制到代码同级目录下

import xml.etree.ElementTree as ET
import matplotlib.pyplot as plt
import matplotlib.image as mpimg # mpimg 用于读取图片
xml_path='000012.xml'
tree = ET.parse(xml_path)
root=tree.getroot()
for object in root.findall('object'):
    a = []
    a.append(object.find('name').text)
    a.append(int(object.find('bndbox').find('xmin').text))
    a.append(int(object.find('bndbox').find('ymin').text))
    a.append(int(object.find('bndbox').find('xmax').text))
    a.append(int(object.find('bndbox').find('ymax').text))

pic = mpimg.imread('000012.jpg')
fig, ax = plt.subplots()
ax.add_patch(plt.Rectangle((a[1],a[2]),a[3]-a[1],a[4]-a[2],fill=False,edgecolor='r', linewidth=2.5))
plt.imshow(pic)
plt.show()