XML便签显示在图片上

不想学习！！！

于 2021-07-11 11:36:59 发布

阅读量529

点赞数

分类专栏：深度学习入门案例

本文链接：https://blog.csdn.net/weixin_43864650/article/details/118652373

版权

OpenCV XML 图像处理边界框对象检测

关键词由CSDN通过智能技术生成

深度学习入门案例专栏收录该内容

5 篇文章

订阅专栏

import os, cv2
import xml.etree.ElementTree as ET
from matplotlib import pyplot as plt

xmls_path = './NEU-val/label/'
imgs_path = './NEU-val/images/'

for images in os.listdir(imgs_path):
    oriname = images.rstrip('.jpg')
    img_path = os.path.join(imgs_path, oriname+'.jpg')
    img = cv2.imread(img_path)

    xml_path = os.path.join(xmls_path, oriname+'.xml')
    tree = ET.parse(xml_path)
    root = tree.getroot()
    for object in root.findall('object'):
        object_name = object.find('name').text
        Xmin = int(object.find('bndbox').find('xmin').text)
        Ymin = int(object.find('bndbox').find('ymin').text)
        Xmax = int(object.find('bndbox').find('xmax').text)
        Ymax = int(object.find('bndbox').find('ymax').text)
        color = (0, 0, 255)
        cv2.rectangle(img, (Xmin, Ymin), (Xmax, Ymax), color, 2)
        font = cv2.FONT_HERSHEY_SIMPLEX
        cv2.putText(img, object_name, (Xmin, Ymin-7), font, 0.5, color, 2)
        cv2.imshow('Image', img)
    cv2.waitKey(0)