xml转化成txt

最新推荐文章于 2024-12-31 17:40:38 发布

转载最新推荐文章于 2024-12-31 17:40:38 发布 · 3.3k 阅读

收录于

常用代码

当前文章被收录于：

常用代码

7 篇文章 0 人学习

订阅专栏查看详情

当前文章被以下社区和专栏收录：

本文介绍了一种将XML文件转换为TXT格式的Python脚本，主要用于解析XML中的对象边界框坐标，转换后的TXT文件用于机器视觉任务的数据标注。脚本读取指定目录下的所有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'
print(file_save)
file_txt = outdir + '/' + file_save
#file_txt=os.path.join(outdir,file_save)
print('**************************')
print(file_txt)
#f_w = open(file_txt,'w')
f_w = open(file_txt,'w',encoding='utf-8')
print('**********************************')

# 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
width = int(xx)-int(xn)
print(width)
height = int(yx)-int(yn)
print(height)
f_w.write('1:'+' ')
f_w.write(xn+' '+yn+' '+str(width)+' '+str(height)+' ')
f_w.write('\n')

indir='C:/Users/Administrator/Desktop/temp/xml' #xml目录
outdir='C:/Users/Administrator/Desktop/temp/txt' #txt目录

xml_to_txt(indir,outdir)