批量将xml文件转换为txt文件

最新推荐文章于 2024-05-16 11:41:50 发布

liujun_Pleasure

最新推荐文章于 2024-05-16 11:41:50 发布

阅读量321

点赞数 4

分类专栏：文件处理文章标签： xml

本文链接：https://blog.csdn.net/liujun221/article/details/136131479

版权

文件处理专栏收录该内容

6 篇文章 0 订阅

订阅专栏

本文介绍了一个Python脚本，用于将目标检测数据集中的XML文件中的标签信息转换为TXT文件，便于进一步训练模型，包括标签名、边界框坐标及其在图像中的比例信息。

摘要由CSDN通过智能技术生成

将目标检测数据集中的xml标签转换为txt文件，方便训练

import os
import xml.etree.ElementTree as ET

dirpath = './1/'  # 原始XML文件的目录
newdir = './3/'  # 存放转换后TXT文件的目录

if not os.path.exists(newdir):
    os.makedirs(newdir)

for xml_file in os.listdir(dirpath):
    xml_path = os.path.join(dirpath, xml_file)
    tree = ET.parse(xml_path)
    root = tree.getroot()
    filename = root.find('filename').text

    with open(os.path.join(newdir, xml_file.split('.')[0] + '.txt'), 'w') as txt_file:
        for obj in root.findall('object'):
            label = obj.find('name').text
            bbox = obj.find('bndbox')
            xmin = float(bbox.find('xmin').text)
            ymin = float(bbox.find('ymin').text)
            xmax = float(bbox.find('xmax').text)
            ymax = float(bbox.find('ymax').text)

            width = float(root.find('size').find('width').text)
            height = float(root.find('size').find('height').text)

            x_center = (xmin + xmax) / (2 * width)
            y_center = (ymin + ymax) / (2 * height)
            w = (xmax - xmin) / width
            h = (ymax - ymin) / height

            txt_file.write(f"{label} {x_center} {y_center} {w} {h}\n")

print('转换完成！')