PASCAL VOC数据集生成train.txt、test.txt、tainval.txt、val.txt

最新推荐文章于 2023-02-05 12:05:10 发布

Ooo。

最新推荐文章于 2023-02-05 12:05:10 发布

阅读量5.3k

点赞数 6

分类专栏：深度学习文章标签： VOC数据集制作

本文链接：https://blog.youkuaiyun.com/xue_csdn/article/details/96448447

版权

Python3代码：

import os
import random 
 
xmlfilepath=r'/home/huadian/darknet/scripts/VOCdevkit/VOC2007/Annotations'
saveBasePath=r"/home/huadian/darknet/scripts/VOCdevkit/VOC2007/ImageSets/Main/"
 
trainval_percent=0.5
train_percent=0.5
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)  
 
print("train and val size",tv)
print("traub suze",tr)
ftrainval = open(os.path.join(saveBasePath,'trainval.txt'), 'w')  
ftest = open(os.path.join(saveBasePath,'test.txt'), 'w')  
ftrain = open(os.path.join(saveBasePath,'train.txt'), 'w')  
fval = open(os.path.join(saveBasePath,'val.txt'), 'w')  
 
for i  in list:  
    name=total_xml[i][:-4]+'\n'  
    if i in