赞
踩
- # 文件名称 :roxml_to_dota.py
- # 功能描述 :把rolabelimg标注的xml文件转换成dota能识别的xml文件,
- # 再转换成dota格式的txt文件
- # 把旋转框 cx,cy,w,h,angle,转换成四点坐标x1,y1,x2,y2,x3,y3,x4,y4
- import os
- import xml.etree.ElementTree as ET
- import math
-
- def edit_xml(xml_file,dotaxml_file):
- """
- 修改xml文件
- :param xml_file:xml文件的路径
- :return:
- """
- tree = ET.parse(xml_file)
- objs = tree.findall('object')
- for ix, obj in enumerate(objs):
- x0 = ET.Element("x0") # 创建节点
- y0 = ET.Element("y0")
- x1 = ET.Element("x1")
- y1 = ET.Element("y1")
- x2 = ET.Element("x2")
- y2 = ET.Element("y2")
- x3 = ET.Element("x3")
- y3 = ET.Element("y3")
- # obj_type = obj.find('bndbox')
- # type = obj_type.text
- # print(xml_file)
-
- if (obj.find('robndbox') == None):
- obj_bnd = obj.find('bndbox')
- obj_xmin = obj_bnd.find('xmin')
- obj_ymin = obj_bnd.find('ymin')
- obj_xmax = obj_bnd.find('xmax')
- obj_ymax = obj_bnd.find('ymax')
- xmin = float(obj_xmin.text)
- ymin = float(obj_ymin.text)
- xmax = float(obj_xmax.text)
- ymax = float(obj_ymax.text)
- obj_bnd.remove(obj_xmin) # 删除节点
- obj_bnd.remove(obj_ymin)
- obj_bnd.remove(obj_xmax)
- obj_bnd.remove(obj_ymax)
- x0.text = str(xmin)
- y0.text = str(ymax)
- x1.text = str(xmax)
- y1.text = str(ymax)
- x2.text = str(xmax)
- y2.text = str(ymin)
- x3.text = str(xmin)
- y3.text = str(ymin)
- else:
- obj_bnd = obj.find('robndbox')
- obj_bnd.tag = 'bndbox' # 修改节点名
- obj_cx = obj_bnd.find('cx')
- obj_cy = obj_bnd.find('cy')
- obj_w = obj_bnd.find('w')
- obj_h = obj_bnd.find('h')
- obj_angle = obj_bnd.find('angle')
- cx = float(obj_cx.text)
- cy = float(obj_cy.text)
- w = float(obj_w.text)
- h = float(obj_h.text)
- angle = float(obj_angle.text)
- obj_bnd.remove(obj_cx) # 删除节点
- obj_bnd.remove(obj_cy)
- obj_bnd.remove(obj_w)
- obj_bnd.remove(obj_h)
- obj_bnd.remove(obj_angle)
-
- x0.text, y0.text = rotatePoint(cx, cy, cx - w / 2, cy - h / 2, -angle)
- x1.text, y1.text = rotatePoint(cx, cy, cx + w / 2, cy - h / 2, -angle)
- x2.text, y2.text = rotatePoint(cx, cy, cx + w / 2, cy + h / 2, -angle)
- x3.text, y3.text = rotatePoint(cx, cy, cx - w / 2, cy + h / 2, -angle)
-
- # obj.remove(obj_type) # 删除节点
- obj_bnd.append(x0) # 新增节点
- obj_bnd.append(y0)
- obj_bnd.append(x1)
- obj_bnd.append(y1)
- obj_bnd.append(x2)
- obj_bnd.append(y2)
- obj_bnd.append(x3)
- obj_bnd.append(y3)
-
- tree.write(dotaxml_file, method='xml', encoding='utf-8') # 更新xml文件
-
-
- # 转换成四点坐标
- def rotatePoint(xc, yc, xp, yp, theta):
- xoff = xp - xc;
- yoff = yp - yc;
- cosTheta = math.cos(theta)
- sinTheta = math.sin(theta)
- pResx = cosTheta * xoff + sinTheta * yoff
- pResy = - sinTheta * xoff + cosTheta * yoff
- return str(int(xc + pResx)), str(int(yc + pResy))
-
-
- def totxt(xml_path,out_path):
-
- # 想要生成的txt文件保存的路径,这里可以自己修改
-
- files = os.listdir(xml_path)
- for file in files:
-
- tree = ET.parse(xml_path + os.sep + file)
- root = tree.getroot()
-
- name = file.strip('.xml')
- output = out_path + name + '.txt'
- file = open(output, 'w')
-
- objs = tree.findall('object')
- for obj in objs:
- cls = obj.find('name').text
- box = obj.find('bndbox')
- x0 = int(float(box.find('x0').text))
- y0 = int(float(box.find('y0').text))
- x1 = int(float(box.find('x1').text))
- y1 = int(float(box.find('y1').text))
- x2 = int(float(box.find('x2').text))
- y2 = int(float(box.find('y2').text))
- x3 = int(float(box.find('x3').text))
- y3 = int(float(box.find('y3').text))
- file.write("{} {} {} {} {} {} {} {} {} 0\n".format(x0, y0, x1, y1, x2, y2, x3, y3, cls))
- file.close()
- print(output)
-
-
- if __name__ == '__main__':
- # -----**** 第一步:把xml文件统一转换成旋转框的xml文件 ****-----
- roxml_path = "./roxml" # 目录下保存的是需要转换的xml文件
- dotaxml_path = './dotaxml'
- out_path = './txt/'
- filelist = os.listdir(roxml_path)
- for file in filelist:
- edit_xml(os.path.join(roxml_path, file), os.path.join(dotaxml_path, file))
-
- # -----**** 第二步:把旋转框xml文件转换成txt格式 ****-----
- totxt(dotaxml_path, out_path)
-
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。