Python实现xml格式转txt格式的示例代码
目录
1、前言2、分析xml、txt数据3、转换过程4、最后结果对比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数据。
代码解析:
import os import xml.etree.ElementTree as ET import io find_path = "./train/" #xml所在的文件 savepath="./labels/" #保存文件 class Voc_Yolo(object): def __init__(self, find_path): self.find_path = find_path def Make_txt(self, outfile): out = open(outfile,"w") print("创建成功:{}".format(outfile)) return out def Work(self, count): #找到文件路径 for root, dirs, files in os.walk(self.find_path): #找到文件目录中每一个xml文件 for file in files: #记录处理过的文件 count += 1 #输入、输出文件定义 input_file = find_path + file outfile = savepath+file[:-4]+".txt" #新建txt文件,确保文件正常保存 out = self.Make_txt(outfile) #分析xml树,取出w_image、h_image tree=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中的四个数据 for obj in root.iter("object"): #将类型提取出来,不同目标类型不同,本文仅有一个类别->0 classname=obj.find("name").text cls_id = classname xmlbox=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_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 #文件写入 out.write(str(cls_id)+" "+str(x_center)+" "+str(y_center)+" "+str(w)+" "+str(h)+"\n") out.close() return count if __name__ == "__main__": data = Voc_Yolo(find_path) number = data.Work(0) print(number)
4、最后结果对比
创建成功
与真实数据对比误差很小
到此这篇关于Python实现xml格式转txt格式的示例代码的文章就介绍到这了,更多相关Python xml转txt内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家!
X 关闭
X 关闭
- 15G资费不大降!三大运营商谁提供的5G网速最快?中国信通院给出答案
- 2联想拯救者Y70发布最新预告:售价2970元起 迄今最便宜的骁龙8+旗舰
- 3亚马逊开始大规模推广掌纹支付技术 顾客可使用“挥手付”结账
- 4现代和起亚上半年出口20万辆新能源汽车同比增长30.6%
- 5如何让居民5分钟使用到各种设施?沙特“线性城市”来了
- 6AMD实现连续8个季度的增长 季度营收首次突破60亿美元利润更是翻倍
- 7转转集团发布2022年二季度手机行情报告:二手市场“飘香”
- 8充电宝100Wh等于多少毫安?铁路旅客禁止、限制携带和托运物品目录
- 9好消息!京东与腾讯续签三年战略合作协议 加强技术创新与供应链服务
- 10名创优品拟通过香港IPO全球发售4100万股 全球发售所得款项有什么用处?