YOLOv5训练怎么只有一个类别啊…

第二次跑YOLOv5了，这次不仅项目体量小还蹭到了N卡设备，心想终于不用让cpu强顶一个晚上了，结果训练完3个类里只有一个……

在loading数据的时候显示全部完成了

但为什么最后跑完了只剩下这点……

python版本是3.11，cuda是12.2如下图

有哪位遇到过吗？这不会是版本问题吧……难道我要重新配置运行环境了吗

1. 你是否加了--single-cls标志？这个标志会将你所有的类别当做一个类来训练
2. 你的数据集当中只有这一个类别

【相关推荐】

• 这篇博客: 全网最全最新的YOLOv5模型训练测试以及多端部署教学内容中的 2.2查询是否支持CUDA安装 部分也许能够解决你的问题, 你可以仔细阅读以下内容或跳转源博客中阅读:
  

  进入：

  https://developer.nvidia.com/cuda-gpus ：

  

  

  从上图中，可以看到我本机的独立显卡是支持CUDA安装的，计算力是6.1。

如果你已经解决了该问题, 非常希望你能够分享一下解决方案, 写成博客, 将相关链接放在评论区, 以帮助更多的人 ^-^