如何提高目标检测模型的置信度,提高mAP

首先引用https://blog.csdn.net/Nire_Yeyu/article/details/105403220

1.调小score_threshold

这种方法属于见效快,但是治标不治本。在验证阶段直接调小score_threshold,可以将mAP快速提升,但是实际上你的模型预测的结果,置信度还是很低。

2.调小batch_size

将原本的batch_size调小,同时增大learning_rate,batch_size可以尝试慢慢调小,因为batch_size小,训练速度会变慢,batch_size大训练速度会相对快一些。
原理是假设你正在看一本书,书的总页数是固定的(相当于数据集的大小)batch_size越大,相当于你每次同时看batch_size页的书,虽然learning_rate较低,即每次看的较慢,但是最终花费的时间短,理解的东西会比较片面,导致阅读质量下降,即预测结果置信度很低,即评估只能通过设置很小的score_threshold,来提升mAP。
所以解决方法是,调小的batch_size,同时增大learning_rate。即每次看较少数量的内容,但是提高阅读速度,总花费时间较长,但理解的东西比较深刻,即最终模型训练的结果也会较好,评估阶段置信度较高。

3.如果是YOLOV3训练

  1. 我遇到的问题是,前期precision很低,recall相对较高,查看结果是预测结果太多了,我的解决方法是提高nms阶段的score_threshold,原本0.01我调整到0.1又调到0.05,最终发现0.05效果较好。
  2. 加更,nms阶段的score_threshold可以继续增加,也可以提升一部分的mAP
  3. 更换优化器,即optimizer_name,可尝试替换其他优化器,查看效果

总结

上述是我将kmeans应用到YOLOV3中,更换自己的anchors时,mAP一直很低,自己多次调整之后,总结的经验。加油自己!离成功又近了一步~

评论 12
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值