目标检测：XML文件矩形框在图片上的简单可视化

代码如下：
需要改的就是开头的四行
可视化结果只有框，没有类别
用于验证
自己写的，亲测可用

import xml.etree.ElementTree as ET
import os
import cv2# ******************************************
src_XML_dir = r'annotations'  # xml源路径
src_IMG_dir = r'images'  # IMG原路径
IMG_format = '.jpg'    # IMG格式
out_dir = 'out'  # 输出路径
# ******************************************if not os.path.exists(out_dir):os.makedirs(out_dir)
xml_file = os.listdir(src_XML_dir)  # 只返回文件名称,带后缀for each_XML in xml_file:  # 遍历所有xml文件# 读入IMGxml_FirstName = os.path.splitext(each_XML)[0]img_save_file = os.path.join(out_dir, xml_FirstName+IMG_format)img_src_path = os.path.join(src_IMG_dir, xml_FirstName+IMG_format)img = cv2.imread(img_src_path)# 解析XMLeach_XML_fullPath = src_XML_dir + '/' + each_XML  # 每个xml文件的完整路径tree = ET.parse(each_XML_fullPath)  # ET.parse()内要为完整相对路径root = tree.getroot()  # 类型为element# 画框for obj in root.findall('object'):if obj.find('bndbox'):bndbox = obj.find('bndbox')xmin = int(bndbox.find('xmin').text)xmax = int(bndbox.find('xmax').text)ymin = int(bndbox.find('ymin').text)ymax = int(bndbox.find('ymax').text)cv2.rectangle(img=img,pt1=(xmin,ymin),pt2=(xmax,ymax),color=(255,0,0),thickness=2)cv2.imwrite(filename=img_save_file, img=img)print('保存结果{}'.format(xml_FirstName))

目标检测：XML文件矩形框在图片上的简单可视化相关推荐

目标检测入门之矩形框IOU计算
1. 引言在目标检测领域中,我们经常用IOU来衡量检测框和标注真实框之间的重叠程度,那么究竟该如何计算IOU呢? 闲话少说,我们直接进入今天的主题- 2. 什么是IOU? IOU(交并比 Inter ...
目标检测xml文件提取
提取xml文件的目标名,xmin,ymin,xmax,ymax,在图上把框框出来并保存. 使用try except避免xml文件中可能存在的标注错误. xml文件提取: 使用xml.dom.minid ...
目标检测之将bbox绘制到图片上
绘制bbox 文件夹内容其中train1里面放的是待绘制的图片 train1_crop.json里面保存的是train1文件夹里数据的信息,三个指标:name,defect_name,bbox 例如 ...
python图片识别拉框_python目标检测给图画框,bbox画到图上并保存案例
我就废话不多说了,还是直接上代码吧! import os import xml.dom.minidom import cv2 as cv ImgPath = 'C:/Users/49691/Deskt ...
根据xml文件找到对应的图片文件
'''根据xml文件找到对应的图片文件代码说明: 这段小程序的作用就是根据labelimage标注软件生成的xml文件来找到对应的图片. 例如:我们的图片数据有100张,其中有90张图片里有我们想标 ...
夏侯南溪搭建目标检测模型——文件结构设计篇
1 前言此文章是夏侯南溪搭建目标检测模型--文件夹结构设计篇- 2 文件夹命名使用小写字母命名,单词之间用短线"-"进行连接,例如: plane-detection 3 项目文 ...
CVPR2019目标检测论文看点：并域上的广义交
CVPR2019目标检测论文看点:并域上的广义交 Generalized Intersection over Union Generalized Intersection over Union: A ...
七牛云 vue 图片上传简单解说，js 上传文件图片
七牛云 vue 图片上传简单解说,js 上传文件图片一.七牛云简介首次使用七牛云存储进行项目的图片存储,整了一上午才整明白,这些官方的教程把明白人也给说糊涂了,文档很不规范. 七牛云有免费的使用额 ...
目标检测标签文件txt转成xml
最近在用ppyolo训练好的模型对新采集的数据进行标记,再人工微调,减少从头打标签的时间,但是推理保存的结果都是txt格式的,想要在labelimg中可视化,那就需要将txt转换成xml. 以下代码即 ...

目标检测：XML文件矩形框在图片上的简单可视化

目标检测：XML文件矩形框在图片上的简单可视化相关推荐

最新文章

热门文章