python读取xml信息和图片，用cv2标注图片

最新推荐文章于 2023-03-21 21:46:57 发布

naomicui

最新推荐文章于 2023-03-21 21:46:57 发布

阅读量4.7k

点赞数 5

分类专栏： pytorch 文章标签： xml cv2 python

本文链接：https://blog.csdn.net/qq_40279508/article/details/81204989

版权

import os
import xml.dom.minidom
import cv2 as cv

ImgPath = '/home/newbee/CODEs/CSRNet-pytorch-master/train/images/'
AnnoPath = '/home/newbee/CODEs/CSRNet-pytorch-master/train/xmls/'

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 = cv.imread(imgfile)

    filenamelist = collection.getElementsByTagName("filename")
    filename = filenamelist[0].childNodes[0].data
    print(filename)
    #得到标签名为object的信息
    objectlist = collection.getElementsByTagName("object")

    for objects in objectlist:
        #每个object中得到子标签名为name的信息
        namelist = objects.getElementsByTagName(