【学习笔记】随机按比例分配数据集小脚本

本文介绍了一个用于神经网络训练数据集的随机划分脚本,旨在保证数据的随机性和充分的特征学习。脚本按照7:2:1的比例自动将数据集划分为训练、验证和测试集。
摘要由CSDN通过智能技术生成

 自己用了一阵,发现来回复制真的很慢,现在不推荐使用这个脚本……自己实名制Diss我自己……


最近频繁使用神经网络训练数据集,目前使用的Tensorflow-Enet需要数据集结构为

├── test
├── testannot
├── train
├── trainannot
├── val
└── valannot

手动分配的话,很难做到随机性。这样就不能保证使用充分的特征来让神经网络学习。因此自己写了一个小脚本,按照训练集、验证集、测试集7:2:1的比例进行划分,脚本如下:

'''
This script is used to ramdomly assign image and
annotation image into test part,train part and 
val part.
@author:Kang Hao
2019.05.16
'''
import os
import random
import shutil

#=============================FILE PATH=========================
  • 3
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值