把数据集NWPU VHR-10转成pascal voc的格式

最新推荐文章于 2025-05-30 09:33:39 发布

原创

最新推荐文章于 2025-05-30 09:33:39 发布 · 3.6k 阅读

16 ·

CC 4.0 BY-SA版权

该博客详细介绍了如何将数据集NWPU VHR-10转换为PASCAL VOC的标准标注格式，包括读取txt文件获取图像区域坐标，了解图像的宽和高，以及生成符合PASCAL VOC规范的xml文件。

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

把数据集NWPU VHR-10转成pascal voc的格式，
1.需要先读取txt文件获得左上角和右下角的坐标。
2.读取图片的宽和高
3.写入xml文件

from lxml.etree import Element,SubElement,tostring
from xml.dom.minidom import parseString
import xml.dom.minidom
import os
import sys
from PIL import Image

#把txt中的内容写进xml
def deal(path):
    files=os.listdir(path)#列出所有文件
    for file in files:
        filename=os.path.splitext(file)[0]#分割出文件名
        #print(filename)
        sufix=os.path.splitext(file)[1]#分割出后缀
        if sufix=='.txt':
            xmins=[]
            ymins=[]
            xmaxs=[]
            ymaxs=[]
            names=[]
            num,xmins,ymins,xmaxs,ymaxs,names=readtxt(file)
            dealpath=path+"/"+filename+".xml"
            filename=filename+'.jpg'
            with open(dealpath,'w') as f:
                writexml(dealpath,filename,num,xmins,ymins,xmaxs,ymaxs,names)

#读取图片的高和宽写入xml        
def dealwh(path):
    files=os.listdir(path)#列出所有文件
    for file in files:
        filename=os.path.splitext(file)[0]#分割出文件名
        sufix=os.path.splitext(file)[1]#分割出后缀
        if sufix