实用小程序（1）之读取xml并保存为txt文本

最新推荐文章于 2025-06-21 09:33:50 发布

原创

最新推荐文章于 2025-06-21 09:33:50 发布 · 760 阅读

标签

#python

收录于

本文介绍了如何使用Python编写小程序，将XML标注文件的类别信息读取并转换为TXT文本格式，适用于Yolo系列算法训练时的数据预处理。主要包括两个步骤：获取训练图片的前缀名保存为TXT和读取XML文件类别。

实用小程序（1）之xml转txt

在这里插入图片描述

这段时间在看学习yolo系列算法，在训练自己的训练集时，需要读取标注的xml文件的类别并保存为txt文本格式，在网上找了很久也没有发现相关的代码，所以就自己写了一个需要的同学自取啦。

（1）获取训练图片的前缀名并保存为txt

import os
import random

xmlfilepath = 'D:\\project\\windows_yolov3\\train_img\\train_image'  # 这里写自己图片保存的路径
txtsavepath = 'D:\\project\\windows_yolov3\\train_img\\name.txt'               #生成的txt文本保存路径
total_xml = os.listdir(xmlfilepath)

num = len(total_xml)
list = range(num)
ftrainval = open(txtsavepath, 'w'