YOLOV5训练数据集P、R、mAP等均为0的解决方案!!!【速进】
可以识别到label,但是P,R,map这些地方都是0,如果确保数据原图和标签是匹配的,或者上传的文件也不是损坏的,那就有可能是梯度的问题了。如果你的代码跟我的有点不一样,没关系是因为yolov5的作者一直在维护升级代码,找到对应位置把默认的SGD优化器改成Adam就可以啦!yolov5默认使用的是SGD优化器,所以我们要改成Adam优化器,Adam优化器也是yolov5封装好的,很简单。如果你的
文章共407字 · 阅读需要大约2分钟
一键AI生成摘要,助你高效阅读
问答
·
问题描述
可以识别到label,但是P,R,map这些地方都是0,如果确保数据原图和标签是匹配的,或者上传的文件也不是损坏的,那就有可能是梯度的问题了。
首先感谢这个哥,我的神啊,快奔溃的时候,偶然瞥到的评论区大哥救了我!!!一起喊,谢谢大哥!!!
解决方案
yolov5默认使用的是SGD优化器,所以我们要改成Adam优化器,Adam优化器也是yolov5封装好的,很简单
如果你的程序跟我的一样的话,参数是这样的
def train()是这样的:
那就直接命令行输入命令加个’–adam’,就相当于指定Adam优化器啦~
python train.py --adam
训练成功!
如果你的代码跟我的有点不一样,没关系是因为yolov5的作者一直在维护升级代码,找到对应位置把默认的SGD优化器改成Adam就可以啦!
宝子们,加油!!!冲冲冲
更多推荐
已为社区贡献1条内容
所有评论(0)