目标检测标签文件xml格式转化为txt格式

原创

已于 2022-10-11 19:09:50 修改 · 1.1k 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#目标检测 #xml

于 2022-10-11 19:09:15 首次发布

本文介绍了一个将XML格式的标注文件转换为TXT格式的Python脚本。该脚本能够解析XML文件中的目标对象位置，并将其转换为YOLOv3等算法所需的标准化坐标格式，方便进行物体检测任务的数据预处理。

import os
from xml.dom import minidom


def xml2txt(xml_path, txt_save_path, labels_list):
    index_label = list(range(len(labels_list)))
    voc_names_dic = dict(zip(labels_list, index_label))
    dir_files = os.listdir(xml_path)
    for item in dir_files:
        doc = minidom.parse(os.path.join(xml_path, item))
        name = doc.getElementsByTagName("filename")[0].firstChild.data  # image_name
        size = doc.getElementsByTagName("size")[0]
        width = int(size.getElementsByTagName("width")[0].firstChild.data)  # width
        height = int(size.getElementsByTagName("height")[0].firstChild.data)  # height
        objects = doc.getElementsByTagName("object")
        label_txt = os.path.join(txt_save_path, name[:-4] + ".txt")
        with open(label_txt, 'w') as file:
            for obj in objects:
                cls = obj.getElementsByTagName("name")[0].firstChild.data  # class_name
                box = obj.getElementsByTagName("bndbox")[