yolo格式（txt）数据集转VOC(xml)

最新推荐文章于 2025-05-13 21:07:34 发布

菜鸟小_pu

最新推荐文章于 2025-05-13 21:07:34 发布

阅读量8.9k

点赞数 25

分类专栏：目标检测Tools 文章标签：深度学习目标检测

本文链接：https://blog.youkuaiyun.com/qq_43161211/article/details/122958255

版权

yolov5训练的数据标签格式为txt格式，有时候会有一些需求需要把txt格式的数据集替换成xml格式的数据集。

1.首先讲一下数据的格式，首先按下图要求创建目录结构。

JPEGImages为图片数据所在的目录
YOLO为yolo格式的标签数据所在目录
Annotations为生成的voc格式数据标签目录（程序运行前这是一个空目录）

2.把对应文件放到对应文件夹下

3.执行yolo_to_voc.py:

from xml.dom.minidom import Document
import os
import cv2
 
 
# def makexml(txtPath, xmlPath, picPath):  # txt所在文件夹路径，xml文件保存路径，图片所在文件夹路径
def makexml(picPath, txtPath, xmlPath):  # txt所在文件夹路径，xml文件保存路径，图片所在文件夹路径
    """此函数用于将yolo格式txt标注文件转换为voc格式xml标注文件
    在自己的标注图片文件夹下建三个子文件夹，分别命名为picture、txt、xml
    """
    dic = {'0': "hat",  # 创建字典用来对类型进行转换
           '1': "person",  # 此处的字典要与自己的classes.txt文件中的类对应，且顺序要一致
           }
    files = os.listdir(txtPath)
    for i, name in enumerate(files):
        xmlBuilder = Document()
        annotation = xmlBuilder.createElement("annotation")  # 创建annotation标签
        xmlBuilder.appendChild(annotation)
        txtFile = open(txtPa