用yolov5自建数据集,训练一个模型,出现如下图所示的问题(run pip install wandb to automatically track and visualize YOLOv5 runs (RECOMMENDED))

 多方面查找资料之后,我发现这种问题的解决办法是

1.在图片数据集建立的时候,以前模型的图片没有删除掉,我是选中了以前的图片,粘贴了现在需要的图片,所以图片数量搞错了,train和test都是这样,然后我又去检查了一下标签数据集labels,发现数量也不对,把图片和标签数据重新修改之后发现还是报错。

2.然后发现自己创建的yaml文件的数量nc忘记填了,

nc填上之后,项目就能愉快的运行了。

总结一下

主要是图片和标签的数量是否正确

另外yaml文件的数目以及种类是否对应正确。 

Logo

GitCode 天启AI是一款由 GitCode 团队打造的智能助手,基于先进的LLM(大语言模型)与多智能体 Agent 技术构建,致力于为用户提供高效、智能、多模态的创作与开发支持。它不仅支持自然语言对话,还具备处理文件、生成 PPT、撰写分析报告、开发 Web 应用等多项能力,真正做到“一句话,让 Al帮你完成复杂任务”。

更多推荐