PASCAL VOC数据集训练集、验证集、测试集的划分和提取
发布日期:2022-03-11 10:18:54 浏览次数:34 分类:技术文章

本文共 1451 字,大约阅读时间需要 4 分钟。

1、训练集、验证集、测试集按比例精确划分

#数据集划分import osimport random root_dir='./park_voc/VOC2007/' ## 0.7train 0.1val 0.2testtrainval_percent = 0.8train_percent = 0.7xmlfilepath = root_dir+'Annotations'txtsavepath = root_dir+'ImageSets/Main'total_xml = os.listdir(xmlfilepath) num = len(total_xml)  # 100list = range(num)tv = int(num*trainval_percent)  # 80tr = int(tv*train_percent)  # 80*0.7=56trainval = random.sample(list, tv)train = random.sample(trainval, tr) ftrainval = open(root_dir+'ImageSets/Main/trainval.txt', 'w')ftest = open(root_dir+'ImageSets/Main/test.txt', 'w')ftrain = open(root_dir+'ImageSets/Main/train.txt', 'w')fval = open(root_dir+'ImageSets/Main/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()

2、训练集、验证集和测试集提取(只给出trian文件的提取方法)

# -*- coding:UTF-8 -*-import shutil f_txt = open('D:\dataset\VOCdevkit\split\VOC2007\ImageSets\Main\\trainval.txt', 'r')f_train = 'D:\dataset\VOCdevkit\VOC2007\\train' context = list(f_txt)for imagename in context:    imagename = imagename[0:6]    imagename = imagename + '.jpg'    imagepath = 'D:\dataset\VOCdevkit\VOC2007\JPEGImages\\'+ imagename    shutil.copy(imagepath,f_train)    # 删除训练集和验证集,剩余图片为测试集    # os.remove(imagepath) #处理Annotations同理只需将.jpg改为.xml

参考:

转载地址:https://blog.csdn.net/weicao1990/article/details/107082865 如侵犯您的版权,请留言回复原文章的地址,我们会给您删除此文章,给您带来不便请您谅解!

上一篇:darket53与yolov3
下一篇:Python3.5.2的pip3升级教程(解决:ImportError: cannot import name 'main')

发表评论

最新留言

路过,博主的博客真漂亮。。
[***.116.15.85]2024年03月06日 13时05分46秒

关于作者

    喝酒易醉,品茶养心,人生如梦,品茶悟道,何以解忧?唯有杜康!
-- 愿君每日到此一游!

推荐文章

php中带?错误,参考-此错误在PHP中意味着什么? 2019-04-21
php生成链接列表,根据URL链接和抛文本生成链接<a>标签的PHP函数 2019-04-21
matlab里inline定义矩阵,Matlab中的inline函数_matlab中inline函数 2019-04-21
php html标签自定义属性,浅谈JS读取DOM对象(标签)的自定义属性 2019-04-21
如何使用matlab的siso,利用Matlab内建程式SISODesignTool完成系统分析(Matlab61)开启.PDF... 2019-04-21
php 实现 model层,Thinkhphp5控制器调用的Model层的方法总结 2019-04-21
matlab6.0序列号,MFC软件获取USB设备的制造商、产品、序列号 2019-04-21
matlab中多边形滤波器,几种常见空间滤波器MATLAB实现 2019-04-21
matlab fminimax 例子,Matlab应用实例(8)—fminimax 2019-04-21
php://filter利用条件,浅谈php://filter技巧 2019-04-21
mplayer-php,mplayer+smplayer 前后端播放器安装 2019-04-21
oracle昨日时间,。。今日,昨日,上周,本月,本年,按时间统计总金额 2019-04-21
php验证卡号,PHP验证信用卡卡号是否正确函数 2019-04-21
mpvue微信小程序动画_推荐两个微信小程序开发框架 2019-04-21
固态硬盘分为哪几种_零基础玩转固态硬盘 深度排雷 买SSD掌握这些就够了 2019-04-21
调python返回图片_Python异常处理,3个好习惯分享给你 2019-04-21
15拆解_收藏:15款劲芯微芯片无线充产品拆解 2019-04-21
弹出u盘_都说:U盘直接拔出不会丢失文件,“安全弹出”形同虚设,对吗? 2019-04-21
怎么查看elementui版本_2021新年 Vue3.0 + Element UI 尝鲜小记 2019-04-21
adreno630gpu参数_小米8搭载Adreno 630图形处理器 比荣耀play上的GPU Turbo更成熟 2019-04-21