将Yolo格式数据集的txt标签转化为Pascal格式数据集的xml标签

最新推荐文章于 2024-04-23 21:47:47 发布

穗虔居士

最新推荐文章于 2024-04-23 21:47:47 发布

阅读量5k

点赞数 1

分类专栏：工具安装文章标签：深度学习 python

本文链接：https://blog.csdn.net/suiqianjushi/article/details/123920391

版权

目的

将Yolo格式数据集的txt标签转化为Pascal格式数据集的xml标签

Yolo格式数据集txt标签：

转化后的 Pascal格式数据集xml标签：

代码

1、新建yolo_convert_voc.py文件，将以下代码复制进去保存。注意：需要根据自己需要修改dic里对应于yolo格式class标签的内容；

from xml.dom.minidom import Document
import os
import cv2


def makexml(txtPath, xmlPath, picPath):  # txt所在文件夹路径，xml文件保存路径，图片所在文件夹路径
    """此函数用于将yolo格式txt标注文件转换为voc格式xml标注文件

    在自己的标注图片文件夹下建三个子文件夹，分别命名为picture、txt、xml
    """
    dic = {#'0': "Bus",  # 创建字典用来对类型进行转换
           #'1': "Microbus",  # 此处的字典要与自己的classes.txt文件中的类对应，且顺序要一致
           #'2': "Minivan",
           #'3': "Sedan",
           #'4': "SUV",
           #'5': "Truck",
            '0': "tricycle",
            '1': "sedan",
            '2': "motorcycle",
            '3': "mixer truck",
            '4': "bicycle",
            '5': "SUV",
            '6': "minibus",
            '7': "slag truck",
            '8': "police car",
            '9': "bus",
            '10