本文主要是介绍数据标注:PascalVOC模式到YOLO模式的一键转化,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!
import os
import xml.etree.ElementTree as ET
from decimal import Decimaldirpath = 'E:\\0911-0951最后一个文件夹\\20190215-211313 {3D675E7F-B913-41B0-B915-9381A662A919}(SHDT-0916(A))\\ZXB_LC01D\\xml' # 原来存放xml文件的目录
newdir = 'E:\\0911-0951最后一个文件夹\\20190215-211313 {3D675E7F-B913-41B0-B915-9381A662A919}(SHDT-0916(A))\\ZXB_LC01D\\new_label' # 修改label后存放txt的目录if not os.path.exists(newdir):os.makedirs(newdir)for fp in os.listdir(dirpath):root = ET.parse(os.path.join(dirpath, fp)).getroot()xmin, ymin, xmax, ymax = 0, 0, 0, 0sz = root.find('size')width = float(sz[0].text)height = float(sz[1].text)filename = root.find('filename').textprint(fp)with open(os.path.join(newdir, fp.split('.')[0] + '.txt'), 'a+') as f:for child in root.findall('object'): # 找到图片中的所有框sub = child.find('bndbox') # 找到框的标注值并进行读取sub_label = child.find('name')xmin = int(sub[0].text)ymin = int(sub[1].text)xmax = int(sub[2].text)ymax = int(sub[3].text)try: # 转换成yolov3的标签格式,需要归一化到(0-1)的范围内x_center = Decimal(str(round(float((xmin + xmax) / (2 * width)),6))).quantize(Decimal('0.000000'))y_center = Decimal(str(round(float((ymin + ymax) / (2 * height)),6))).quantize(Decimal('0.000000'))w = Decimal(str(round(float((xmax - xmin) / width),6))).quantize(Decimal('0.000000'))h = Decimal(str(round(float((ymax - ymin) / height),6))).quantize(Decimal('0.000000'))print(str(x_center) + ' ' + str(y_center)+ ' '+str(w)+ ' '+str(h))if sub_label.text == 'FrontBolt':f.write(' '.join([str(0), str(x_center), str(y_center), str(w), str(h) + '\n']))if sub_label.text == 'SideBolt':f.write(' '.join([str(1), str(x_center), str(y_center), str(w), str(h) + '\n']))if sub_label.text == 'Part':f.write(' '.join([str(2), str(x_center), str(y_center), str(w), str(h) + '\n']))except ZeroDivisionError:print(filename, '的 width有问题')# with open(os.path.join(newdir, fp.split('.')[0] + '.txt'), 'a+') as f:# f.write(' '.join([str(2), str(x_center), str(y_center), str(w), str(h) + '\n']))
这篇关于数据标注:PascalVOC模式到YOLO模式的一键转化的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!