自己用了一阵,发现来回复制真的很慢,现在不推荐使用这个脚本……自己实名制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=========================