数据集划分问题

[xiaohuan1208]

您好？请问数据集划分相关代码在那一部分，请指教。不胜感激

[selous123]

@xiaohuan1208
数据集划分是通过配置文件中的配置实现的：
https://github.com/selous123/yolov3-pytorch-custom/blob/master/data/rubbish.data

具体生成方式可以参考库：
ultralytics/yolov3#1570

[xiaohuan1208]

@selous123
华为垃圾检测分类数据集是没有划分训练集和测试集的。我想知道作者是通过什么样的方式来划分训练集和测试集的。比如，按照类别数量等比抽取这样的方式，项目中是否有相关代码？

[selous123]

相关代码我已经找不到了, 这个库里面也没有相关的代码.
基本逻辑就是随机分,8:2的比例左右.
一般随机分就可以达到按照类别数量等比抽取这样的效果

[SpongeBab]

对于coco，官网就给了训练集和验证集了，yolo本身是没有数据集划分的。一般是torch.data.random_split().好像是这个。