【AI目标检测】VOC格式数据集转换为DOTA类型数据集

原创已于 2022-03-24 20:56:54 修改 · 3.1k 阅读

17 ·

CC 4.0 BY-SA版权

文章标签：

#python #计算机视觉 #目标检测

于 2022-03-21 20:57:19 首次发布

该代码实现了从VOC格式的XML标注文件到DOTA格式的转换。VOC XML文件包含了图像的尺寸信息、物体类别及其旋转边界框坐标，而DOTA格式则简化为四点坐标及类别名。转换过程中，通过旋转点的算法计算新的边界框坐标，并写入到TXT文件中。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

由于目前的imglabel画出来框都是voc类型的xml文件：

<annotation>
	<folder>rotate_jueyuanzi_zip</folder>
	<filename>0004.jpg</filename>
	<path>D:\Desktop\rotate\dataset\rotate_jueyuanzi_zip\0004.jpg</path>
	<source>
		<database>Unknown</database>
	</source>
	<size>
		<width>1037</width>
		<height>778</height>
		<depth>3</depth>
	</size>
	<segmented>0</segmented>
	<object>
		<name>comp</name>
		<pose>Unspecified</pose>
		<truncated>0</truncated>
		<difficult>0</difficult>
		<robndbox>
			<cx>453.9988</cx>
			<cy>159.4623</cy>
			<w>44.3132</w>
			<h>106.04</h>
			<angle>4.963185</angle>
		</robndbox>
		<extra/>
	</object>
	<object>
		<name>comp</name>
		<pose>Unspecified</pose>
		<truncated>0</truncated>
		<difficult>0</difficult>
		<robndbox>
			<cx>548.4988</cx>
			<cy>159.4623</cy>
			<w>35.867</w>
			<h>98.7147</h>
			<angle>4.963185</angle>
		</robndbox>
		<extra/>
	</object>
	<object>
		<name>comp</name>
		<pose>Unspecified</pose>
		<truncated>0</truncated>
		<difficult>0</difficult>
		<robndbox>
			<cx>486.0786</cx>
			<cy>304.3616</cy>
			<w>45.8548</w>
			<h>99.6275</h>
			<angle>0.81</angle>
		</robndbox>
		<extra/>
	</object>
	<object>
		<name>comp</name>
		<pose>Unspecified</pose>
		<truncated>0</truncated>
		<difficult>0</difficult>
		<robndbox>
			<cx>585.526</cx>
			<cy>326.5931</cy>
			<w>46.7375</w>
			<h>102.784</h>
			<angle>0.81</angle>
		</robndbox>
		<extra/>
	</object>
</annotation>

想要转换成为DOTA类型的数据：

513.9 363.3 622.1 466.1 578.5 511.9 470.3 409.1 comp 0
280.0 198.9 411.9 159.6 427.1 210.7 295.3 250.1 comp 0
404.4 198.2 538.1 165.5 550.7 217.4 417.1 250.1 comp 0
373.9 332.3 482.1 435.1 438.5 480.9 330.3 378.1 comp 0

转换代码：

import os
import xml.etree.ElementTree as ET
import math


def voc_to_dota(xml_path, xml_name):
    txt_name = xml_name[:-4] + '.txt'
    txt_path = xml_path + '/txt_label'
    if not os.path.exists(txt_path):
        os.makedirs(txt_path)
    txt_file = os.path.join(txt_path, txt_name)
    file_path = os.path.join(xml_path, file_list[i])
    tree = ET.parse(os.path.join(file_path))
    root = tree.getroot()
    # print(root[6][0].text)
    with open(txt_file, "w+", encoding='UTF-8') as out_file:
        # out_file.write('imagesource:null' + '\n' + 'gsd:null' + '\n')
        for obj in root.findall('object'):
            name = obj.find('name').text
            difficult = obj.find('difficult').text
            # print(name, difficult)
            robndbox = obj.find('robndbox')
            cx = float(robndbox.find('cx').text)
            cy = float(robndbox.find('cy').text)
            w = float(robndbox.find('w').text)
            h = float(robndbox.find('h').text)
            angle = float(robndbox.find('angle').text)
            # print(cx, cy, w, h, angle)
            p0x, p0y = rotatePoint(cx, cy, cx - w / 2, cy - h / 2, -angle)
            p1x, p1y = rotatePoint(cx, cy, cx + w / 2, cy - h / 2, -angle)
            p2x, p2y = rotatePoint(cx, cy, cx + w / 2, cy + h / 2, -angle)
            p3x, p3y = rotatePoint(cx, cy, cx - w / 2, cy + h / 2, -angle)
            data = str(p0x) + " " + str(p0y) + " " + str(p1x) + " " + str(p1y) + " " + \
                   str(p2x) + " " + str(p2y) + " " + str(p3x) + " " + str(p3y) + " "
            data = data + name + " " + difficult + "\n"
            out_file.write(data)


# 转换成四点坐标
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
    # pRes = (xc + pResx, yc + pResy)
    # 保留一位小数点
    return float(format(xc + pResx, '.1f')), float(format(yc + pResy, '.1f'))
    # return xc + pResx, yc + pResy


if __name__ == '__main__':
    root_path = '../annotation'
    file_list = os.listdir(root_path)
    for i in range(0, len(file_list)):
        if ('.xml' in file_list[i]) or ('.XML' in file_list[i]):
            voc_to_dota(root_path, file_list[i])
            print('----------------------------------------{}{}----------------------------------------'
                  .format(file_list[i], ' has Done!'))
        else:
            print(file_list[i] + ' is not xml file')