从XML文件制作VOC数据集ImageSets下的txt文件

最新推荐文章于 2021-12-21 17:07:07 发布

Tom Hardy

最新推荐文章于 2021-12-21 17:07:07 发布

阅读量2.5k

点赞数 2

分类专栏：深度学习

深度学习专栏收录该内容

113 篇文章 53 订阅

订阅专栏

直接上源代码，配置好路径直接可以使用

import os
import random

trainval_percent = 0.8  #作为trainval_percent的比例，在整个数据集，可以修改
train_percent = 0.8 #用于训练的数据的比例，可以修改
xmlfilepath = 'F:/pic/VOC Data/Annotations'
txtsavepath = 'F:/pic/VOC Data/ImageSets/Main'
total_xml = os.listdir(xmlfilepath)

num=len(total_xml)
list=range(num)
tv=int(num*trainval_percent)
tr=int(tv*train_percent)
trainval= random.sample(list,tv)
train=random.sample(trainval,tr)

ftrainval = open(txtsavepath+'/trainval.txt', 'w')
ftest = open(txtsavepath+'/test.txt', 'w')
ftrain = open(txtsavepath+'/train.txt', 'w')
fval = open(txtsavepath+'/val.txt', 'w')

for i  in list:
    name=total_xml[i][:-4]+'\n'
    if i in trainval:
        ftrainval.write(name)
        if i in train:
            ftrain.write(name)
        else:
            fval.write(name)
    else:
        ftest.write(name)

ftrainval.close()
ftrain.close()
fval.close()
ftest .close()