将txt文件转为yolo格式的xml文件

最新推荐文章于 2024-05-09 11:57:43 发布

无人机怎么飞

最新推荐文章于 2024-05-09 11:57:43 发布

阅读量1.2k

点赞数 2

CC 4.0 BY-SA版权

文章标签：大数据 xml python 计算机视觉

本文链接：https://blog.csdn.net/coulson/article/details/125778467

这篇博客介绍了如何使用Python将TXT格式的图像标签转换为YOLO所需的XML格式，以便于目标检测。作者提供了一个脚本，该脚本读取TXT文件中的类别标签和边界框信息，然后创建XML文件，包含必要的图像尺寸、类别和边界框坐标。脚本处理多目标检测，并能处理同一图像中的多个对象。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

最近在参加一个图像处理的比赛，赛中遇到了一些问题，记录一下。

需要将txt文件转化为yolo格式的xml文件，就是可以在labelImg中看见目标检测框的那种xml文件。

代码如下

# .txt-->.xml
# ! /usr/bin/python
# -*- coding:UTF-8 -*-
import os
import cv2


def txt_to_xml(txt_path, img_path, xml_path):
    # 1.字典对标签中的类别进行转换
    dict = {'0': "holothurian",
            '1': "echinus",
            '2': "scallop",
            '3': "starfish",
            '4': "waterseed"
            }
    # 2.找到txt标签文件夹
    files = os.listdir(txt_path)
    # 用于存储 "老图"
    pre_img_name = ''
    # 3.遍历文件夹
    for i, name in enumerate(files):
        # 许多人文件夹里有该文件，默认的也删不掉，那就直接pass
        if name == "desktop.ini":
            continue
        print(name)
        # 4.打开txt
        txtFile = open(txt_path + name)
        # 读取所有内容
        txtList = txtFile.readlines()
        # 读取图片名称
        img_name = name.split(".")[0]
        pic = cv2.imread(img_path + img_name + ".jpg")
        # 获取图像大小信息
        Pheight, Pwidth, Pdepth = pic.shape
        # 5.遍历txt文件中每行

最低0.47元/天解锁文章

200万优质内容无限畅学