人工智能数据集划分脚本划分训练集和测试集

划分训练集和测试集本脚本食用方法十分简单原理是提取目录中的文件名称随机打乱放到对应的数组中后续再对数组中的内容进行对比提取存在标签的图片最后将图片和标签划分到训练集验证集测试集.默认情况下我提供了一个自动选择根路径的函数如果你不做任何修改可以直接按要求选择未划分的数据集即可.各个集的划分情况如下训练集70验证集20测试集10