0

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

显示错误

4

2 回答 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的自定义数据集重现您的错误,并且我的训练成功开始:

YOLOv5 训练

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

于 2021-07-10T12:30:26.880 回答