无法按照 YOLOv5 的 Roboflow 教程训练数据集

问题描述

我正在尝试按照 Roboflow 上的 youtube 教程在我的自定义数据集上训练 YOlov5。但是,我在尝试训练数据集时不断收到错误消息。

Error shown

解决方法

我尝试使用来自 roboflow 和 YOLOv5 Colab notebook 的自定义数据集重现您的错误,并且我的训练成功开始:

YOLOv5 training

根据您的错误屏幕截图,该训练似乎正在为训练、有效和测试寻找多个数据集。这可能意味着您将多个数据集复制到笔记本环境中,并且模型不确定您要使用哪个数据集进行训练。我建议 (1) 重新启动笔记本内核并仅使用 Roboflow 代码片段导入您的数据集一次 (2) 在 Roboflow 中调查数据集以验证您是否拥有模型预期的训练、有效和测试拆分。如果您没有此拆分,则 Roboflow 确实允许您在创建导出时重新平衡训练、验证、测试和拆分。

,

用他们的 colab notebook 关注他们的博客,你就会明白:

https://blog.roboflow.com/how-to-train-yolov5-on-a-custom-dataset/