将xml文件中的标注画到图片数据集中并保存

最新推荐文章于 2023-09-25 18:58:04 发布

ljw1173489866

最新推荐文章于 2023-09-25 18:58:04 发布

阅读量617

点赞数 2

文章标签： xml python 开发语言

本文链接：https://blog.csdn.net/ljw1173489866/article/details/125738829

版权

import cv2
import os
import os.path
import xml.etree.cElementTree as ET

def draw(image_path, xml_path, root_saved_path):
    "图片根据标注画框"
    src_img_path = image_path
    src_ann_path = xml_path
    for file in os.listdir(src_ann_path):
        file_name, suffix = os.path.splitext(file)
        if suffix == '.xml':
            xml_path = os.path.join(src_ann_path,file)
            image_path = os.path.join(src_img_path,file_name+'.jpg')
            img = cv2.imread(image_path)
            tree = ET.parse(xml_path)
            root = tree.getroot()

            for obj in root.iter('object'):
                name = obj.find('name').text
                xml_box = obj.find('bndbox')
                x1 = int(xml_box.find('xmin').text)
                x2 = int(xml_box.find('xmax').text)
                y1 = int(xml_box.find('ymin').text)