Python+opencv读取xml文件并显示

最新推荐文章于 2024-05-31 13:31:56 发布

小鹏聊智驾

最新推荐文章于 2024-05-31 13:31:56 发布

阅读量2.2k

点赞数 2

分类专栏： DL工程技巧文章标签： xml python opencv

本文链接：https://blog.youkuaiyun.com/chiyukunpeng/article/details/108274642

版权

代码

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,