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 如侵犯您的版权,请留言回复原文章的地址,我们会给您删除此文章,给您带来不便请您谅解!
发表评论
最新留言
路过,博主的博客真漂亮。。
[***.116.15.85]2024年03月06日 13时05分46秒
关于作者
喝酒易醉,品茶养心,人生如梦,品茶悟道,何以解忧?唯有杜康!
-- 愿君每日到此一游!
推荐文章
php中带?错误,参考-此错误在PHP中意味着什么?
2019-04-21
php生成链接列表,根据URL链接和抛文本生成链接<a>标签的PHP函数
2019-04-21
php html标签自定义属性,浅谈JS读取DOM对象(标签)的自定义属性
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