Python+opencv读取xml文件并显示

1. 代码

import os
import xml.dom.minidom
import cv2

ImgPath = './train-A/image/'
AnnoPath = './train-A/box/'
imagelist = os.listdir(ImgPath)

for image in imagelist:
    image_pre, ext = os.path.splitext(image)  # 分离文件名与扩展名
    imgfile = ImgPath + image
    xmlfile = AnnoPath + image_pre + '.xml'

    # 打开xml文档
    DOMTree = xml.dom.minidom.parse(xmlfile)
    # 得到文档元素对象
    collection = DOMTree.documentElement
    # 读取图片
    img = cv2.imread(imgfile)

    filenamelist = collection.getElementsByTagName("frame")
    filename = filenamelist[0].childNodes[0].data
    cv2.putText(img,filename,(10,20), cv2.FONT_HERSHEY_COMPLEX, 0.7, (0, 255, 0), thickness=2)
    img_size_list=collection.getElementsByTagName('size')
    for img_size in img_size_list:
        img_width_list = img_size.getElementsByTagName('width')
        img_height_list = img_size.getElementsByTagName('height')
        img_width = img_width_list[0].childNodes[0].data
        img_height = img_height_list[0].childNodes[0].data
        cv2.putText(img, img_width, (10, 40), cv2.FONT_HERSHEY_COMPLEX, 0.7, (0, 255, 0), thickness=2)
        cv2.putText(img, img_height, (10, 60), cv2.FONT_HERSHEY_COMPLEX, 0.7, (0, 255, 0), thickness=2)

    # 得到标签名为object的信息
    objectlist = collection.getElementsByTagName("object")
    for objects in objectlist:
        # 每个object中得到子标签名为name的信息
        namelist = objects.getElementsByTagName('name')
        objectname = namelist[0].childNodes[0].data
        bndbox = objects.getElementsByTagName('bndbox')
        for box in bndbox:
            x1_list = box.getElementsByTagName('xmin')
            x1 = int(x1_list[0].childNodes[0].data)
            y1_list = box.getElementsByTagName('ymin')
            y1 = int(y1_list[0].childNodes[0].data)
            x2_list = box.getElementsByTagName('xmax')
            x2 = int(x2_list[0].childNodes[0].data)
            y2_list = box.getElementsByTagName('ymax')
            y2 = int(y2_list[0].childNodes[0].data)
            cv2.rectangle(img, (x1, y1), (x2, y2), (255, 0, 0), thickness=2)
            cv2.putText(img, objectname, (x1, y1), cv2.FONT_HERSHEY_COMPLEX, 0.7, (0, 255, 0), thickness=2)

    cv2.imshow('img', img)
    cv2.waitKey(0)
    cv2.destroyAllWindows()

2. xml样例

<?xml version="1.0" ?>
<annotation>
	<frame>000001</frame>
	<object>
		<name>holothurian</name>
		<bndbox>
			<xmin>712</xmin>
			<ymin>415</ymin>
			<xmax>838</xmax>
			<ymax>573</ymax>
		</bndbox>
	</object>
	<object>
		<name>echinus</name>
		<bndbox>
			<xmin>129</xmin>
			<ymin>940</ymin>
			<xmax>300</xmax>
			<ymax>1080</ymax>
		</bndbox>
	</object>
	<object>
		<name>echinus</name>
		<bndbox>
			<xmin>344</xmin>
			<ymin>894</ymin>
			<xmax>467</xmax>
			<ymax>1002</ymax>
		</bndbox>
	</object>
	<object>
		<name>echinus</name>
		<bndbox>
			<xmin>653</xmin>
			<ymin>799</ymin>
			<xmax>792</xmax>
			<ymax>924</ymax>
		</bndbox>
	</object>
	<object>
		<name>echinus</name>
		<bndbox>
			<xmin>342</xmin>
			<ymin>540</ymin>
			<xmax>468</xmax>
			<ymax>673</ymax>
		</bndbox>
	</object>
	<object>
		<name>echinus</name>
		<bndbox>
			<xmin>1668</xmin>
			<ymin>251</ymin>
			<xmax>1833</xmax>
			<ymax>412</ymax>
		</bndbox>
	</object>
	<size>
		<width>1920</width>
		<height>1080</height>
	</size>
</annotation>