问题描述
可以识别到label,但是P,R,map这些地方都是0,如果确保数据原图和标签是匹配的,或者上传的文件也不是损坏的,那就有可能是梯度的问题了。
在这里插入图片描述
首先感谢这个哥,我的神啊,快奔溃的时候,偶然瞥到的评论区大哥救了我!!!一起喊,谢谢大哥!!!
在这里插入图片描述
解决方案
yolov5默认使用的是SGD优化器,所以我们要改成Adam优化器,Adam优化器也是yolov5封装好的,很简单
如果你的程序跟我的一样的话,参数是这样的
在这里插入图片描述
def train()是这样的:
在这里插入图片描述

那就直接命令行输入命令加个’–adam’,就相当于指定Adam优化器啦~

python train.py --adam

训练成功!
在这里插入图片描述

如果你的代码跟我的有点不一样,没关系是因为yolov5的作者一直在维护升级代码,找到对应位置把默认的SGD优化器改成Adam就可以啦!

宝子们,加油!!!冲冲冲

Logo

旨在为数千万中国开发者提供一个无缝且高效的云端环境,以支持学习、使用和贡献开源项目。

更多推荐