错误：yolov5训练后，部分类别检测的P、R和mAP为0

最新推荐文章于 2024-10-31 15:29:34 发布

weixin_45478423

最新推荐文章于 2024-10-31 15:29:34 发布

阅读量5.4k

点赞数 4

文章标签： python 算法

本文链接：https://blog.youkuaiyun.com/weixin_45478423/article/details/129032705

版权

文章讲述了在使用Yolov5进行目标检测训练时遇到的问题，即某些类别的检测指标P、R和mAP为0。问题出在数据集的标签文件转换和配置文件的classes定义不一致。作者发现，在未指定类别时，数据划分会按字母顺序分配编号，但配置文件中手动指定了类别顺序。为解决此问题，需确保data.yaml中name的顺序与转换文件classes一致，重新运行后，模型的检测结果得到了改善。