DOTA数据集转成PASCAL VOC的格式

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

原创

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

38 ·

CC 4.0 BY-SA版权

本文档介绍了如何将DOTA数据集转换为PASCAL VOC格式，详细说明了转换过程，并提供了修改后的源代码链接。

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

原作者github代码https://github.com/DetectionTeamUCAS/RRPN_Faster-RCNN_Tensorflow/tree/master/data/io
但在我这跑不通，部分细节改了改。
将DOTA数据集转成PASCAL VOC的格式

import os
from xml.dom.minidom import Document
from xml.dom.minidom import parse
import xml.dom.minidom
import numpy as np
import csv
import cv2
import string

def WriterXMLFiles(filename, path, box_list, label_list, w, h, d):

    # dict_box[filename]=json_dict[filename]
    doc = xml.dom.minidom.Document()
    root = doc.createElement('annotation')
    doc.appendChild(root)

    foldername = doc.createElement("folder")
    foldername.appendChild(doc.createTextNode("JPEGImages"))
    root.appendChild(foldername)

    nodeFilename = doc.createElement('filename')
    nodeFilename.appendChild(doc.createTextNode(filename))
    root.appendChild(nodeFilename)

    pathname = doc.createElement("path")
    pathname.appendChild(doc.createTextNode("xxxx"))
    root.appendChild(pathname)

    sourcename=doc.createElement("source")

    databasename = doc.createElement("database")
    databasename.appendChild(doc.createTextNode("Unknown"))
    sourcename.appendChild(databasename)

    annotationname = doc.createElement("annotation")
    annotationname.appendChild(doc.createTextNode("xxx"))
    sourcename