Python Yolov5数据集转Voc数据集

qq_52217283

于 2022-08-02 21:52:24 发布

阅读量432

点赞数

本文链接：https://blog.youkuaiyun.com/qq_52217283/article/details/126126884

版权

YOLOv5 VOC格式数据集转换 XML文件图像标注

关键词由优快云通过智能技术生成

把yoloV5识别出的结果文本转换至voc格式数据集

代码:

import os
import os.path
from PIL import Image
from xml.dom.minidom import Document


def write_xml(tmp, image, w, h, obj_bud, wx_ml, classes):
    doc = Document()
    # owner
    annotation = doc.createElement('annotation')
    doc.appendChild(annotation)
    # owner
    folder = doc.createElement('folder')
    annotation.appendChild(folder)
    folder_txt = doc.createTextNode("VOC2005")
    folder.appendChild(folder_txt)

    filename = doc.createElement('filename')
    annotation.appendChild(filename)
    filename_txt = doc.createTextNode(image)
    filename.appendChild(filename_txt)
    # ones#
    source = doc.createElement('source')
    annotation.appendChild(source)

    database = doc.createElement('database')
    source.appendChild(database)
    database_txt = doc.createTextNode("The VOC2005 Database")
    database.appendChild