【模式识别4】YOLO目标检测数据集xml格式转txt格式

YOLO 目标检测数据集xml格式转txt格式

- 1. 转换前的xml格式
- 2. xml格式转txt格式代码
- - 2.1 源代码
  - 2.2 需要修改的地方
- 3. 转换后的txt格式

代码资源：voc2txt.py

xml_5">1. 转换前的xml格式

如果我们使用LabelImg工具标注数据集，生成的xml文件如下：
在这里插入图片描述
xml中各标签含义：

<object>：检测的目标
<name>：检测对象 / 类别
<pose>：拍摄角度（Left：左//Right：右//Unspecified：未指明）
<truncated>：是否被截断（0 代表检测目标完整//1 代表检测目标被截断）
<difficult>：是否容易被检测（0 代表容易检测）
<bndbox> ：bounding-box（包含左上角 x,y 坐标<xmin>、<ymin>和右下角 x,y 坐标<xmax>、<ymax>) (单位：像素）

xmltxt_15">2. xml格式转txt格式代码

2.1 源代码

# 导入相关库
import os
from lxml import etree
from tqdm import tqdm


def voc2txt():
    # 获取xml文件夹下的所有xml文件名，存入列表
    xmls_list = os.listdir(xmls_ori_path)
    for xml_name in tqdm(xmls_list):
        # 打开写入文件
        txt_name = xml_name.replace('xml', 'txt')

        f = open(os.path.join(txts_save_path, txt_name), 'w')  # 代开待写入的txt文件
        with open(os.path.join(xmls_ori_path, xml_name), 'rb') as fp:
            # 开始解析xml文件
            xml = etree.HTML(fp.read())
            width = int(xml.xpath('//size/width/text()')[0])
            height = int(xml.xpath('//size/height/text()')[0])
            # 获取对象标签
            obj = xml.xpath('//object')
            for each in obj:
                name = each.xpath("./name/text()")[0]
                classes = dic[name]
                xmin = int(each.xpath('./bndbox/xmin/text()')[0])
                xmax = int(each.xpath('./bndbox/xmax/text()')[0])
                ymin = int(each.xpath('./bndbox/ymin/text()')[0])
                ymax = int(each.xpath('./bndbox/ymax/text()')[0])
                # 归一化
                dw = 1 / width
                dh = 1 / height
                x_center = (xmin + xmax) / 2
                y_center = (ymax + ymin) / 2
                w = (xmax - xmin)
                h = (ymax - ymin)
                x, y, w, h = x_center * dw, y_center * dh, w * dw, h * dh
                # 写入
                f.write(str(classes) + ' ' + str(x) + ' ' + str(y) + ' ' + str(w) + ' ' + str(h) + ' ' + '\n')
        f.close()  # 关闭txt文件


if __name__ == '__main__':
    dic = {'zero': "0",  # 创建字典用来对类型进行转换
           'one': "1",   # 此处的字典要与自己的classes.txt文件中的类对应，且顺序要一致
           'two': "2",
           'three': "3",
           'four': "4",
           'five': "5",
           'six': "6",
           'seven': "7",
           'eight': "8",
           'nine': "9",
           }

    xmls_ori_path = r"xml"         # xml文件所在的文件夹
    txts_save_path = r"txt"         # txt文件所在的文件夹

    os.mkdir(txts_save_path) if not os.path.exists(xmls_ori_path) else None

    voc2txt()

2.2 需要修改的地方

目标的类别
LabelImg标注完会生成“classes.txt”，如下为真实类别名称，从上到下类别代号分别为0、1、2、…

因此，需要建立字典映射，将classes.txt的真实类别名称（如下左列）分别映射到代号0、1、2…（如下右列）
修改数据集路径
修改为自己的xml文件夹路径以及保存的txt文件夹路径即可