我正在尝试按照 Roboflow 上的 youtube 教程在我的自定义数据集上训练 YOLOv5。但是,当我尝试训练我的数据集时,我不断收到错误消息。
问问题
215 次
2 回答
0
用他们的 colab notebook 关注他们的博客,你会明白的:
https://blog.roboflow.com/how-to-train-yolov5-on-a-custom-dataset/
于 2021-07-18T08:11:09.940 回答
0
我尝试使用来自 roboflow 和YOLOv5 Colab notebook的自定义数据集重现您的错误,并且我的训练成功开始:
根据您的错误屏幕截图,训练似乎正在为训练、有效和测试寻找多个数据集。这可能意味着您将多个数据集复制到笔记本环境中,并且模型不确定您要使用哪个进行训练。我建议 (1) 重新启动笔记本内核并仅使用 Roboflow 代码片段导入数据集一次 (2) 调查 Roboflow 中的数据集,以验证您是否拥有模型所期望的训练、有效和测试拆分。如果您没有此拆分,Roboflow 确实使您能够在创建导出时重新平衡训练、验证、测试和拆分。
于 2021-07-10T12:30:26.880 回答