YOLO数据集划分参数trainval_percent 和train_percent的含义

YOLO数据集划分参数trainval_percent 和train_percent的含义

trainval_percent = 0.8
# 训练集与验证集 占全体数据的比例
train_percent = 0.9
# 训练集 占训练集与验证集总体 的比例
...
'''
需要生成以下四个文件,分别是test.txt,train.txt,trainval.txt,val.txt
分别代表测试集、训练集、训练-验证集、验证集
'''
file_trainval = open(txtsavepath + '/trainval.txt', 'w')
file_test = open(txtsavepath + '/test.txt', 'w')
file_train = open(txtsavepath + '/train.txt', 'w')
file_val = open(txtsavepath + '/val.txt', 'w')

本人理解,trainval_percent = 0.8表示训练-验证集占总体的80%,其对应的是trainval.txt文件,测试集占总体的20%,其对应的是test.txt文件。
train_percent = 0.9表示trainval.txt文件中提取90%,组成train.txt文件,其余10%组成val.txt文件

完整划分数据集代码参考https://blog.csdn.net/qq_36756866/article/details/109111065

你可能感兴趣的:(深度学习,人工智能)