目标检测--将xml文件中标签（矩形框）在其原图片上显示并另存

""" 目的：将原图片(img)与其xml(xml)，合成为打标记的图片(labelled)，矩形框标记用红色即可 已有：（1）原图片文件夹(imgs_path)，（2）xml文件夹(xmls_path) 思路：step1: 读取（原图片文件夹中的）一张图片step2: 读取（xmls_path）该图片的xml文件，并获取其矩形框的两个对角顶点的位置step3: 依据矩形框顶点坐标，在该图片中画出该矩形框step4: 图片另存为'原文件名'+'_labelled'，存在‘lablled’文件夹中 """ import os import cv2 as cv import xml.etree.ElementTree as ETdef xml_jpg2labelled(imgs_path, xmls_path, labelled_path):imgs_list = os.listdir(imgs_path)xmls_list = os.listdir(xmls_path)nums = len(imgs_list)for i in range(nums):img_path = os.path.join(imgs_path, imgs_list[i])xml_path = os.path.join(xmls_path, xmls_list[i])img = cv.imread(img_path)labelled = imgroot = ET.parse(xml_path).getroot()objects = root.findall('object')for obj in objects:bbox = obj.find('bndbox')xmin = int(float(bbox.find('xmin').text.strip()))ymin = int(float(bbox.find('ymin').text.strip()))xmax = int(float(bbox.find('xmax').text.strip()))ymax = int(float(bbox.find('ymax').text.strip()))labelled = cv.rectangle(labelled, (xmin, ymin), (xmax, ymax), (0, 0, 255), 1)cv.imwrite('%s%s_labelled.jpg' % (labelled_path, imgs_list[i]), labelled)# cv.imshow('labelled', labelled)# cv.imshow('origin', origin)# cv.waitKey()if __name__ == '__main__':imgs_path = '自定义'xmls_path = '自定义'labelled_path = '自定义'xml_jpg2labelled(imgs_path, xmls_path, labelled_path)

总结

以上是生活随笔为你收集整理的目标检测--将xml文件中标签（矩形框）在其原图片上显示并另存的全部内容，希望文章能够帮你解决所遇到的问题。

如果觉得生活随笔网站内容还不错，欢迎将生活随笔推荐给好友。