标注工具生成的xml文件转为txt格式

最新推荐文章于 2025-07-21 10:42:06 发布

Enjoy_endless

最新推荐文章于 2025-07-21 10:42:06 发布

阅读量1.8w

点赞数 13

CC 4.0 BY-SA版权

分类专栏： Deep learning

本文链接：https://blog.youkuaiyun.com/Enjoy_endless/article/details/80819945

Deep learning 专栏收录该内容

45 篇文章

订阅专栏

本文介绍了一种将LabelImg生成的XML格式标注文件转换为坐标加标签TXT格式的方法，适用于多数网络训练的需求。通过解析XML文件获取标注框坐标，并将其写入TXT文件中。

上一篇文章介绍学习了标注工具labelImg的安装及详细使用，可以看到其标注成功之后生成的为xml格式文件：
这里写图片描述

但是大多数网络训练的时候需要对其进行格式转换，转为：坐标+标签的txt格式类型的，所以如下程序用于实现此功能：

import os
import sys
import xml.etree.ElementTree as ET
import glob

def xml_to_txt(indir,outdir):

    os.chdir(indir)
    annotations = os.listdir('.')
    annotations = glob.glob(str(annotations)+'*.xml')

    for i, file in enumerate(annotations):

        file_save = file.split('.')[0]+'.txt'
        file_txt=os.path.join(outdir,file_save)
        f_w = open(file_txt,'w')

        # actual parsing
        in_file = open(file)
        tree=ET.parse(in_file)
        root = tree.getroot()

        for obj in root.iter('object'):
                current = list()
                name = obj.find('name').text

                xmlbox = obj.find('bndbox')
                xn = xmlbox.find('xmin').text
                xx = xmlbox.find('xmax').text
                yn = xmlbox.find('ymin').text
                yx = xmlbox.find('ymax').text
                #print xn
                f_w.write(xn+' '+yn+' '+xx+' '+yx+' ')
                f_w.write(name.encode("utf-8")+'\n')

indir='/home/zgz/data-xml'   #xml目录
outdir='/home/zgz/data-txt'  #txt目录

xml_to_txt(indir,outdir)