UA-DETRAC数据集转yolo格式.pdf

UA-DETRAC数据集转yolo格式 UA-DETRAC数据集转yolo格式 格式如下所⽰： 转换代码： import os import xml.etree.ElementTree as ET # DET RAC-Train-Annotations-XML内涵xml⽂件 xmlDir=r"D:\ aiduNetdiskDownload\ UA_ DET RAC\ DET RAC-Train-Annotations-XML" # 转换的xml⽂件保存位置 new_dir=r"D:\ aiduNetdiskDownload\ UA_ DET RAC\train\xml" # 转换xml⽂件 def bboxes2xml(folder, img_ name, width, height, gts, xml_save_to) : def bboxes2xml(folder, img_ name, width, height, gts, xml_save_to) : xml_file = open((xml_save_to + '/' + img_ name + '.xml'), 'w') xml_file.write('<annotation>\n') xml_file.write(' <folder>' + folder + '</folder>\n') xml_file.write(' <filename>' + str(img_ name) + '.jpg' + '</filename>\n') xml_file.write(' <size>\n') xml_file.write(' <width>' + str(width) + '</width>\n') xml_file.write(' <height>' + str(height) + '</height>\n') xml_file.write(' <depth>3</depth>\n') xml_file.write(' </size>\n') for gt in gts: xml_file.write(' <object>\n') xml_file.write(' <name>' + str(gt[0]) + '</name>\n') xml_file.write(' <pose>Unspecified</pose>\n') xml_file.write(' <truncated>0</truncated>\n') xml_file.write(' <difficult>0</difficult>\n') xml_file.write(' <bndbox>\n') xml_file.write(' <xmin>' + str(gt[1]) + '</xmin>\n') xml_file.write(' <ymin>' + str(gt[2]) + '</ymin>\n') xml_file.write(' <xmax>' + str(gt[3]) + '</xmax>\n') xml_file.write(' <ymax>' + str(gt[4]) + '</ymax>\n') xml_file.write(' </bndbox>\n') xml_file.write(' </object>\n') xml_file.write('</annotation>') xml_file.close() for xmlNames in os.listdir(xmlDir) : xmlPath=os.path.join(xmlDir,xmlNames) # xml⽂件路径 # print(xmlPath) tree = ET .parse(xmlPath) root = tree.getroot() findall_frames =