此代码将数据集分为:训练+验证,训练集,验证集,测试集,根据实际情况取相对应的文件即可。比如:
若你的训练代码中不包含验证集划分,取训练集,验证集,测试集使用;若包含验证集划分,取训练+验证,测试集使用。
新建一个split_train_test.py文件,用来划分数据集,代码如下:
import os
import random
trainval_percent = 0.9 # 训练+验证
train_percent = 0.9 #训练
xmlfilepath = 'DataSet\\Annotations' # 标注文件路径
txtsavepath = 'DataSet\\Main'

本文提供了一个用于划分数据集的Python代码实例,该代码能够将数据集划分为训练集、验证集和测试集,并可根据需要调整比例。
最低0.47元/天 解锁文章
1242

被折叠的 条评论
为什么被折叠?



