xml_txt

简单将xml数据转换为txt数据&＃xff08;仅当记录&＃xff09;

import xml.etree.ElementTree as ET
import pickle
import os
from os import listdir, getcwd
from os.path import joinsets&＃61;[&＃39;train&＃39;,&＃39;trainval&＃39;]classes &＃61; ["mono"]def convert(size, box):dw &＃61; 1./size[0]dh &＃61; 1./size[1]x &＃61; (box[0] &＃43; box[1])/2.0y &＃61; (box[2] &＃43; box[3])/2.0w &＃61; box[1] - box[0]h &＃61; box[3] - box[2]x &＃61; x*dww &＃61; w*dwy &＃61; y*dhh &＃61; h*dhreturn (x,y,w,h)def convert_annotation(image_id): # 转换这一张图片的坐标表示方式&＃xff08;格式&＃xff09;,即读取xml文件的内容&＃xff0c;计算后存放在txt文件中。in_file &＃61; open(&＃39;annotation/%s&＃39;% image_id)image_id&＃61;image_id.split(&＃39;.&＃39;)[0]out_file &＃61; open(&＃39;labels/%s.txt&＃39;%image_id, &＃39;w&＃39;)tree&＃61;ET.parse(in_file)root &＃61; tree.getroot()size &＃61; root.find(&＃39;size&＃39;)w &＃61; int(size.find(&＃39;width&＃39;).text)h &＃61; int(size.find(&＃39;height&＃39;).text)for obj in root.iter(&＃39;object&＃39;):difficult &＃61; obj.find(&＃39;difficult&＃39;).textcls &＃61; obj.find(&＃39;name&＃39;).textif cls not in classes or int(difficult) &＃61;&＃61; 1:continuecls_id &＃61; classes.index(cls)xmlbox &＃61; obj.find(&＃39;bndbox&＃39;)b &＃61; (float(xmlbox.find(&＃39;xmin&＃39;).text), float(xmlbox.find(&＃39;xmax&＃39;).text), float(xmlbox.find(&＃39;ymin&＃39;).text), float(xmlbox.find(&＃39;ymax&＃39;).text))bb &＃61; convert((w,h), b)out_file.write(str(cls_id) &＃43; " " &＃43; " ".join([str(a) for a in bb]) &＃43; &＃39;\n&＃39;)for image_set in sets:if not os.path.exists(&＃39;labels/&＃39;):os.makedirs(&＃39;labels/&＃39;) # 新建一个 label 文件夹&＃xff0c;用于存放yolo格式的标签文件&＃xff1a;000001.txtimage_ids &＃61; open(&＃39;%s.txt&＃39; % image_set).read().strip().split()# 读取txt文件中 存放的图片的 id&＃xff1a;000001for image_id in image_ids:image_id&＃61;image_id.split(&＃39;/&＃39;)[-1]image_id&＃61;image_id.split(&＃39;.&＃39;)[0]image_id&＃61;image_id&＃43;".xml"convert_annotation(image_id) # 转换这一张图片的坐标表示方式&＃xff08;格式&＃xff09;