如何提高目标检测模型的置信度,提高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一直很低,自己多次调整之后,总结的经验。加油自己!离成功又近了一步~

  • 17
    点赞
  • 123
    收藏
    觉得还不错? 一键收藏
  • 12
    评论
回答: 目标检测中的mAP(mean Average Precision)是一种衡量模型性能的指标。mAP的计算方法相对复杂,但可以简单概括为以下几个步骤。首先,对于每个类别的目标,计算出模型在不同置信度阈值下的精确度(Precision)和召回率(Recall)。然后,根据不同置信度阈值下的精确度和召回率计算出每个类别的AP(Average Precision)。最后,将所有类别的AP取平均得到mAP。 具体计算步骤如下: 1. 对于每个类别的目标,根据置信度从高到低对检测结果进行排序。 2. 根据排序后的结果,计算出每个检测框的精确度和召回率。 3. 根据不同置信度阈值下的精确度和召回率,计算出每个类别的PR曲线。 4. 对于每个类别的PR曲线,计算出其下的AP值。 5. 将所有类别的AP值取平均得到mAP。 需要注意的是,mAP的计算方法可能会有一些细微的差异,具体取决于不同的实现和评估方法。但总体来说,mAP目标检测中常用的性能评估指标,用于衡量模型在多个类别上的检测准确度。\[2\] #### 引用[.reference_title] - *1* *2* *3* [目标检测指标mAP详解](https://blog.csdn.net/qq_38375203/article/details/125391658)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v91^control_2,239^v3^insert_chatgpt"}} ] [.reference_item] [ .reference_list ]

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 12
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值