python文件操作之xml转txt

最新推荐文章于 2024-04-12 22:45:00 发布
最新推荐文章于 2024-04-12 22:45:00 发布
阅读量1.4k 收藏 3
点赞数 1
文章标签： python xml java
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/qq_46454669/article/details/134478596
版权

 在使用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文件的文件名和后缀，其中xmlname[0] = 文件名，xmlname[1] = 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.childNodes[0].data == 'armor_red'

file_txt.write('0 ')

通过调用name.childNodes[0].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()

nice-wyh
关注
  • 1
    点赞
  • 踩
  • 3
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
将xml文件转换为txt格式文件（python）
05-07
将xml文件转换为txt格式文件（python）
TXT格式转xml格式（voc）
03-26
"TXT格式转XML格式（voc）"指的是将纯文本的TXT文件转换为结构化的XML文件，以适应特定的应用场景，例如语音识别（VOC可能指的是Voice Over IP或者Voice Output Control）。XML是一种可扩展标记语言，它允许我们在...
目标检测：如何将VOC标注的xml数据转为YOLO标注的txt格式，且生成classes的txt文件
Henry的博客
01-08 3670
根据VOC标注的xml文件生成YOLO格式的txt文件，且生成yolo需要的类别txt文件
Python——详细解析目标检测xml格式标注转换为txt格式
weixin_53848907的博客
04-12 704
本文简述了目标检测xml格式标注的内容，以及yolo系列模型所需的txt格式标注的内容。并提供了一个简单的，可以将xml格式标注文件转换为txt格式标注文件的python脚本。
xml文件转成txt文件
zqq19980906_的博客
07-01 2641
xml文件转成txt文件。
VOC数据集转换为yolo格式：xml转txt(亲测有用)
pyscl01的博客
08-31 3168
数据集格式转换：xml转txt(亲测有用)的详细教程
XML 转 TXT
u011669976的博客
04-27 2767
import os import xml.etree.ElementTree as ET # xml文件存放目录（不要以\结尾） input_dir = r'C:\Users\Administrator\Desktop\zzz\xml' # 输出txt文件目录（不要以\结尾） out_dir = r'C:\Users\Administrator\Desktop\zzz\txt' class_list = [] # 获取目录所有xml文件 def file_name(input_dir): .
（六）标签格式转换xml转txt
雪剑封心
06-15 3234
标签格式转换xml转txt你也可以修改文件名,把对应的字符串改了也可以。 1.创建文件夹：label_xml 放入xml 2.创建文件夹：label_txt 等待生成txt
【模式识别4】YOLO目标检测数据集xml格式转txt格式
weixin_43387635的博客
04-22 2453
LabelImg标注完会生成“classes.txt”，如下为真实类别名称，从上到下类别代号分别为0、1、2、…因此，需要建立字典映射，将classes.txt的真实类别名称（如下左列）分别映射到代号0、1、2…每行表示（类别，中心x相对坐标，中心y相对坐标，相对宽度、相对高度）修改为自己的xml文件夹路径以及保存的txt文件夹路径即可。
tools_python；xml；mat文件_
10-01
总结，这个压缩包提供的工具展示了Python在数据处理和文件操作上的强大能力，涉及了Python基础、文件I/O、XML解析、MAT文件操作等多个重要知识点，这些都是在进行数据分析、机器学习或计算机视觉项目时经常遇到的...
数据集划分，xml划分，txt划分，xml转换成txt
最新发布
04-24
Python的`xml.etree.ElementTree`库可以解析XML，然后将其写入TXT文件，或者使用XSLT转换来实现这个目的。转换后，TXT文件可用于简单的文本分析、文本挖掘或作为输入到不支持XML格式的工具。 总的来说，这些步骤都...
为了工作方便刚刚写的一个txt转xml的工具
03-31
NULL 博文链接：https://balan326.iteye.com/blog/1344812
txt数据转换为xml数据的小工具
12-28
很多数据的存储都是在txt文档，或者数据库上，这个小工具可以将txt格式的数据库转换为xml类型的数据，然后在我们的程序中使用。注意，需要稍微修改一下代码，思路是很好的！
TXT转化XML
07-31
TXT转化XML，用于讲文档转化啊为XML格式的工具
VOC数据集xml to yolov4需要的txt文件
02-22
总的来说，这些Python脚本构成了一个从VOC XML标注到YOLOv4训练所需TXT格式的转换工具。通过运行这些脚本，我们可以方便地将VOC数据集转换成YOLOv4可以接受的格式，从而可以利用VOC数据集丰富的资源来训练高效的...
python_elementt_xml.zip_python xml_xml解析 python
09-14
`ElementTree`库提供了解析XML文档为元素树（Element Tree）的方法，这使得我们可以遍历和操作XML结构。 1. **解析XML文档**：使用`ElementTree.parse()`或`ET.fromstring()`函数解析XML文件。前者用于读取整个XML...
深度学习——xml格式转txt格式数据集
qq_40231851的博客
04-24 717
【代码】深度学习——xml格式转txt格式数据集。
将xml文件转yolov5训练数据txt标签文件
m0_46935103的博客
08-02 3618
使用正则表达式提取xml文件的关键内容，经算法处理，转化为YOLO训练需要的txt格式数据，完整代码已展示，另提供代码下载
python批量实现labelImg标注的 xml格式数据转换成 txt格式保存
weixin_72965172的博客
12-13 599
【代码】python批量实现labelImg标注的 xml格式数据转换成 txt格式保存。
用python把txt文件转xml文件
03-29
以下是一个简单的Python脚本，它将txt文件转换为XML文件： ```python import xml.etree.ElementTree as ET # 打开txt文件 with open('example.txt', 'r') as f: data = f.readlines() # 创建根元素 root = ET.Element('data') # 添加子元素 for line in data: child = ET.SubElement(root, 'line') child.text = line.strip() # 创建XML文件 tree = ET.ElementTree(root) tree.write('example.xml') ``` 此脚本假定源文本文件名为“example.txt”，并且每行文本都应转换为XML中的一个“line”元素。输出的XML文件名为“example.xml”。您可以根据需要更改这些文件名，并根据需要修改子元素的名称。

“相关推荐”对你有帮助么？

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值