python文件操作之xml转txt

nice-wyh2023-11-19 13:02

在使用yolo进行深度学习训练时，我们所使用的标签文件都是txt格式的，但是有的人使用的标注软件生成的可能是xml文件，那么就需要使用python工具写一个格式转换脚本。

首先导入库，并定义标注的图片地址、生成的标签文件xml地址、存储结果txt的地址。

复制代码 
import cv2
import os
import math
import xml.dom.minidom

path_img = r'/home/pc/下载/qq_file/test/img/'
path_xml = r'/home/pc/下载/qq_file/test/xml/'
path_txt = r'/home/pc/下载/qq_file/test/txt/'

files = os.listdir(path_xml)这句话的files存储了path_xml路径下的所有xml文件

复制代码 
if __name__ == "__main__":
    files = os.listdir(path_xml)
    print("文件数量： " + str(len(files)))
    for file in files:
        print(file)

如上代码的file打印结果就应该为目标文件夹下所有xml文件的文件名，如246.xml。

复制代码 
if __name__ == "__main__":
    files = os.listdir(path_xml)
    print("文件数量： " + str(len(files)))
    for file in files:
        print(file)
        xmlname = file.split('.')
        txtname = xmlname[0] + '.txt'
        imgname = xmlname[0] + '.jpg'

        txtpath = path_txt + txtname
        imgpath = path_img + imgname
        xmlpath = path_xml + xmlname[0] + '.' + xmlname[1]

接下来用" . "来分割xml文件的文件名和后缀，其中xmlname0 = 文件名，xmlname1 = xml。由此可以定义每一个xml、txt、jpg的路径。

复制代码 
if __name__ == "__main__":
    files = os.listdir(path_xml)
    print("文件数量： " + str(len(files)))
    for file in files:
        print(file)
        xmlname = file.split('.')
        txtname = xmlname[0] + '.txt'
        imgname = xmlname[0] + '.jpg'

        txtpath = path_txt + txtname
        imgpath = path_img + imgname
        xmlpath = path_xml + xmlname[0] + '.' + xmlname[1]
        # print(xmlpath)

        file_txt = open(txtpath, 'w+')
        frame = cv2.imread(imgpath)
        pixel_size = frame.shape[0]
        print(pixel_size)

接下来定义file_txt为文件夹下每一个txt文件，并可进行write操作。

复制代码 
# 将解析xml文件的结果存在dom变量下
        dom = xml.dom.minidom.parse(xmlpath)
        # 获取xml文件的根元素
        root = dom.documentElement
        objects = root.getElementsByTagName('object')

        for obj in objects:
            name = obj.getElementsByTagName('name')[0]  # [0]可能代表包含标签数值的一个集合（这里可能包含标签数值,地址等）
            if name.childNodes[0].data == 'armor_red':  # 通过调用name.childNodes[0].data方法获取标签数值
                file_txt.write('0 ')
            elif name.childNodes[0].data == 'full_red':
                file_txt.write('1 ')
            elif name.childNodes[0].data == 'R_red':
                file_txt.write('2 ')
            else:
                print(file + ":label error")

            robodbox = obj.getElementsByTagName('robodbox')
            cx = obj.getElementsByTagName('cx')[0]
            cy = obj.getElementsByTagName('cy')[0]
            w = obj.getElementsByTagName('w')[0]
            h = obj.getElementsByTagName('h')[0]
            angle = obj.getElementsByTagName('angle')[0]
            # print(f"cx = {cx.childNodes[0]}, cy = {cy.childNodes[0].data}, w = {w.childNodes[0].data}, h = {h.childNodes[0].data}, "
            #       f"angle = {angle.childNodes[0].data}")

            temp_cx = str(float(cx.childNodes[0].data) / pixel_size)
            temp_cy = str(float(cy.childNodes[0].data) / pixel_size)
            temp_h = str(float(h.childNodes[0].data) / pixel_size)
            temp_w = str(float(w.childNodes[0].data) / pixel_size)
            temp_angle = str(float(angle.childNodes[0].data) / math.pi*180)

接下来对xml文件进行解析，调用dom = xml.dom.minidom.parse(xmlpath)，将解析xml文件的结果存在dom变量下；root = dom.documentElement，获取xml文件的根元素root；objects = root.getElementsByTagName('object')，获得object；使用for循环遍历每一个object，使用name = obj.getElementsByTagName('name')0来获得每一个object的name，这里0可能代表包含标签数值的一个集合（这里可能包含标签数值,地址等），然后判断

if name.childNodes0.data == 'armor_red'

file_txt.write('0 ')

通过调用name.childNodes0.data方法获取标签数值，写入txt文件作为第一个数值，即"类别"。

复制代码 
            robodbox = obj.getElementsByTagName('robodbox')
            cx = obj.getElementsByTagName('cx')[0]
            cy = obj.getElementsByTagName('cy')[0]
            w = obj.getElementsByTagName('w')[0]
            h = obj.getElementsByTagName('h')[0]
            angle = obj.getElementsByTagName('angle')[0]
            # print(f"cx = {cx.childNodes[0]}, cy = {cy.childNodes[0].data}, w = {w.childNodes[0].data}, h = {h.childNodes[0].data}, "
            #       f"angle = {angle.childNodes[0].data}")

            temp_cx = str(float(cx.childNodes[0].data) / pixel_size)
            temp_cy = str(float(cy.childNodes[0].data) / pixel_size)
            temp_h = str(float(h.childNodes[0].data) / pixel_size)
            temp_w = str(float(w.childNodes[0].data) / pixel_size)
            temp_angle = str(float(angle.childNodes[0].data) / math.pi*180)

            file_txt.write(temp_cx + ' ')
            file_txt.write(temp_cy + ' ')
            file_txt.write(temp_w + ' ')
            file_txt.write(temp_h + ' ')
            file_txt.write(temp_angle + ' ')
            file_txt.write('\n')

类别信息获取后，接下来对矩形框数值进行解析，使用同样的方法获取数据，并进行归一化写入txt。

完整代码如下

复制代码 
import cv2
import os
import math
import xml.dom.minidom

path_img = r'/home/pc/下载/qq_file/test/img/'
path_xml = r'/home/pc/下载/qq_file/test/xml/'
path_txt = r'/home/pc/下载/qq_file/test/txt/'

if __name__ == "__main__":
    files = os.listdir(path_xml)
    print("文件数量： " + str(len(files)))
    for file in files:
        print(file)
        xmlname = file.split('.')
        txtname = xmlname[0] + '.txt'
        imgname = xmlname[0] + '.jpg'

        txtpath = path_txt + txtname
        imgpath = path_img + imgname
        xmlpath = path_xml + xmlname[0] + '.' + xmlname[1]
        # print(xmlpath)

        file_txt = open(txtpath, 'w+')
        frame = cv2.imread(imgpath)
        pixel_size = frame.shape[0]
        # print(pixel_size)

        # 将解析xml文件的结果存在dom变量下
        dom = xml.dom.minidom.parse(xmlpath)
        # 获取xml文件的根元素
        root = dom.documentElement
        objects = root.getElementsByTagName('object')

        for obj in objects:
            name = obj.getElementsByTagName('name')[0]  # [0]可能代表包含标签数值的一个集合（这里可能包含标签数值,地址等）
            if name.childNodes[0].data == 'armor_red':  # 通过调用name.childNodes[0].data方法获取标签数值
                file_txt.write('0 ')
            elif name.childNodes[0].data == 'full_red':
                file_txt.write('1 ')
            elif name.childNodes[0].data == 'R_red':
                file_txt.write('2 ')
            else:
                print(file + ":label error")

            robodbox = obj.getElementsByTagName('robodbox')
            cx = obj.getElementsByTagName('cx')[0]
            cy = obj.getElementsByTagName('cy')[0]
            w = obj.getElementsByTagName('w')[0]
            h = obj.getElementsByTagName('h')[0]
            angle = obj.getElementsByTagName('angle')[0]
            # print(f"cx = {cx.childNodes[0]}, cy = {cy.childNodes[0].data}, w = {w.childNodes[0].data}, h = {h.childNodes[0].data}, "
            #       f"angle = {angle.childNodes[0].data}")

            temp_cx = str(float(cx.childNodes[0].data) / pixel_size)
            temp_cy = str(float(cy.childNodes[0].data) / pixel_size)
            temp_h = str(float(h.childNodes[0].data) / pixel_size)
            temp_w = str(float(w.childNodes[0].data) / pixel_size)
            temp_angle = str(float(angle.childNodes[0].data) / math.pi*180)

            file_txt.write(temp_cx + ' ')
            file_txt.write(temp_cy + ' ')
            file_txt.write(temp_w + ' ')
            file_txt.write(temp_h + ' ')
            file_txt.write(temp_angle + ' ')
            file_txt.write('\n')

        print("ok")
        file_txt.close()
相关推荐
2501_941875282 小时前
从配置中心到动态管理的互联网工程语法演进与多语言实践分享
开发语言·python
hfywmsj2 小时前
广州餐饮铺位招租决策模型：多因子选址系统设计
开发语言·人工智能·python·广州餐饮铺位招租
lhldsg5 小时前
全民健身解决方案小程序开发：从0到1的技术实战
java·数据库·需求分析
jason成都5 小时前
Spring WebFlux 适配达梦新方案｜dm‑r2dbc：Netty 异步传输的实验性 R2DBC 驱动
java·后端·spring
泡海椒5 小时前
内置SPI函数库详解：JQuick-Java Builtin工具类实战用法
java·开发语言·python
hqyjzsb5 小时前
零 AI 项目经验，学 Python 转型 AI 的正确顺序是什么？
开发语言·人工智能·python·算法·职场和发展·数据挖掘·数据分析
辰烨chenye5 小时前
LeetCode Hot 100 题解 · 二分篇
java·算法·leetcode
小羊没烦恼！6 小时前
Hello Web API系列教程——Web API与国际化
java·服务器·前端·javascript·php
sanjiaomao3336 小时前
从论文公式到Python实现：用单元测试校验滑动平均算法
python·算法·单元测试
小荷才露尖尖角，早有蜻蜓立上头7 小时前
过滤器的4种创建方式
java
热门推荐
01国内可直接用、免费额度/永久免费的大模型API清单（含 SiliconFlow、火山、阿里、智谱、百度、Kimi、DeepSeek、DMXAPI 等）02GitHub 镜像站点03如何新建文件夹？ 电脑新建文件夹的4种方法04历年考研数学一、数学二、数学三真题试卷及答案PDF05蜘蛛磁力 搜索引擎大全，如何使用蜘蛛磁力查找磁力链接062026 国产 AI 大模型横评：DeepSeek、通义千问、Kimi、文心一言、星火、豆包谁更能打？07微信历史版本含下载地址（ Windows PC | 安卓 | MAC ）及设置微信不更新08ZCode 周末送额度活动开启：3 亿 Token 免费领取09Agnes AI 免费 API 接入指南：文本、生图、生视频，一套接口全免费10AI 编程 IDE 全景解析 2026：Agent 全面接管开发链路