dota数据集的裁剪并转成xml格式

最新推荐文章于 2025-06-18 15:15:18 发布

原创

最新推荐文章于 2025-06-18 15:15:18 发布 · 3k 阅读

16 ·

CC 4.0 BY-SA版权

本文介绍如何对Dota数据集进行裁剪，并将其转换为XML格式，适用于机器学习和游戏分析。首先，我们将探讨数据预处理的重要性和步骤，接着详细解释裁剪过程，包括选择关键特征和剔除冗余数据。然后，我们讨论使用Python库如pandas和xml.etree.ElementTree来转换数据格式。最后，我们将提供完整的代码示例和注意事项，帮助读者实现自己的数据转换任务。

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

import os
import scipy.misc as misc
from xml.dom.minidom import Document
import numpy as np
import copy, cv2

def save_to_xml(save_path, im_height, im_width, objects_axis, label_name,name):
    im_depth = 0
    object_num = len(objects_axis)
    doc = Document()

    annotation = doc.createElement('annotation')
    doc.appendChild(annotation)

    folder = doc.createElement('folder')
    folder_name = doc.createTextNode('VOC2007')
    folder.appendChild(folder_name)
    annotation.appendChild(folder)

    filename = doc.createElement('filename')
    filename_name = doc.createTextNode(name)
    filename.appendChild(filename_name)
    annotation.appendChild(filename)


    size = doc.createElement('size')
    annotation.appendChild(size)
    width = doc.createElement('width')
    width.appendChild(doc.createTextNode(str(im_width)))
    height = doc.createElement('height')
    height.appendChild(doc.createTextNode(str(im_height)))
    depth = doc.createElement('depth')
    depth.appendChild(doc.createTextNode(str(im_depth)))
    size.appendChild(width)
    size.appendChild(height)
    size.appendChild(depth)
    segmented = doc.createElement('segmented')
    segmented.appendChild(doc.createTextNode('0'))
    annotation.appendChild(segmented)
    for i in range(object_num):
        objects = doc.createElement('object')
        annotation.appendChild(objects)
        object_name = doc.createElement('name')
        object_name.appendChild(doc.createTextNode(label_name[int(objects_axis[i][-1])]))
        objects.appendChild(object_name)
        pose = doc.createElement('pose')
        pose.appendChild(doc.createTextNode('Unspecified'))
        objects.appendChild(pose)
        truncated = doc.createElement('truncated')
        truncated.appendChild(