Python如何实现xml格式转txt格式

魁首哥

作者

Python如何实现xml格式转txt格式

本篇内容介绍了“Python如何实现xml格式转txt格式”的有关知识，在实际案例的操作过程中，不少人都会遇到这样的困境，接下来就让小编带领大家学习一下如何处理这些情况吧！希望大家仔细阅读，能够学有所成！

1、前言

最近学习Yolo v5是遇见了个问题，找的数据集全是xml文件，VOC 的标注是 xml 格式的，而YOLO是.txt格式，那么问题就来了，手动提取肯定是不可能的，那只能借用程序解决咯。

2、分析xml、txt数据

这是xml树形结构

这是txt格式

总结:

1.提取object->name、bndbox->xmin,ymin,xmax,ymin

2.格式转化需要用公式转换

YOLO数据集txt格式：

x_center ：归一化后的中心点x坐标

y_center ：归一化后的中心点y坐标

w:归一化后的目标框宽度

h：归一化后的目标况高度

（此处归一化指的是除以图片宽和高）

VOC数据集xml格式

yolo的四个数据	xml->txt公式
x_center	((x_min+x_max)/2-1)/w_image
y_center	((y_min+y_max)/2-1)/h_image
w	(x_max-x_min)/w_image
h	(y_max-y_min)/h_image

3、转换过程

定义两个文件夹，train放xml数据, labels放txt数据。

代码解析：

importosimportxml.etree.ElementTreeasETimportiofind_path='./train/'#xml所在的文件savepath='./labels/'#保存文件classVoc_Yolo(object):def__init__(self,find_path):self.find_path=find_pathdefMake_txt(self,outfile):out=open(outfile,'w')print("创建成功：{}".format(outfile))returnoutdefWork(self,count):#找到文件路径forroot,dirs,filesinos.walk(self.find_path):#找到文件目录中每一个xml文件forfileinfiles:#记录处理过的文件count+=1#输入、输出文件定义input_file=find_path+fileoutfile=savepath+file[:-4]+'.txt'#新建txt文件，确保文件正常保存out=self.Make_txt(outfile)#分析xml树，取出w_image、h_imagetree=ET.parse(input_file)root=tree.getroot()size=root.find('size')w_image=float(size.find('width').text)h_image=float(size.find('height').text)#继续提取有效信息来计算txt中的四个数据forobjinroot.iter('object'):#将类型提取出来，不同目标类型不同，本文仅有一个类别->0classname=obj.find('name').textcls_id=classnamexmlbox=obj.find('bndbox')x_min=float(xmlbox.find('xmin').text)x_max=float(xmlbox.find('xmax').text)y_min=float(xmlbox.find('ymin').text)y_max=float(xmlbox.find('ymax').text)#计算公式x_center=((x_min+x_max)/2-1)/w_imagey_center=((y_min+y_max)/2-1)/h_imagew=(x_max-x_min)/w_imageh=(y_max-y_min)/h_image#文件写入out.write(str(cls_id)+""+str(x_center)+""+str(y_center)+""+str(w)+""+str(h)+'\n')out.close()returncountif__name__=="__main__":data=Voc_Yolo(find_path)number=data.Work(0)print(number)