【python脚本】python实现xml文件的图片标签可视化

批量实现文件夹下xml文件可视化展现结果。

import xml.dom.minidom
import cv2
import os
import xml.etree.ElementTree as ET"""
该脚本用于目标框可视化
IMAGE_INPUT_PATH：输入图片路径
XML_INPUT_PATH：输入标记框路径
IMAGE_OUTPUT_PATH：生成可视化图片路径
"""
IMAGE_INPUT_PATH = 'xx/data/1'
XML_INPUT_PATH = 'xx/data/3'
IMAGE_OUTPUT_PATH = 'xx/data/2'imglist = os.listdir(IMAGE_INPUT_PATH)
xmllist = os.listdir(XML_INPUT_PATH)for i in range(len(imglist)):# 每个图像全路径image_input_fullname = IMAGE_INPUT_PATH + '/' + imglist[i]xml_input_fullname = XML_INPUT_PATH + '/' + xmllist[i]image_output_fullname = IMAGE_OUTPUT_PATH + '/' + imglist[i]image = cv2.imread(image_input_fullname)# 解析XML文件tree = ET.parse(xml_input_fullname)root = tree.getroot()# 获取图像尺寸width = int(root.find('size/width').text)height = int(root.find('size/height').text)# 遍历所有目标for obj in root.findall('object'):# 获取目标类别和边界框坐标label = obj.find('name').textxmin = int(obj.find('bndbox/xmin').text)ymin = int(obj.find('bndbox/ymin').text)xmax = int(obj.find('bndbox/xmax').text)ymax = int(obj.find('bndbox/ymax').text)# 根据类别选择颜色if label == 'Car':color = (255, 0, 0)  # 蓝色elif label == 'Van':color = (0, 255, 0)  # 绿色                        else:color = (255, 192, 203)  # 不同的颜色# 在图像上画出边界框和类别标签cv2.rectangle(image, (xmin, ymin), (xmax, ymax), color, 2)cv2.putText(image, label, (xmin, ymin - 5), cv2.FONT_HERSHEY_SIMPLEX, 0.5, color, 2)"""# 直接查看生成结果图 cv2.imshow('show', img)cv2.waitKey(0)"""cv2.imwrite(image_output_fullname, image)