修改mmdetection置信度

在使用Mask-RCNN进行图像分割时,遇到图片被识别为多个类别的情况,特别是那些置信度低于60%的类别。通过修改tools/test.py中的--show-score-thr参数,将阈值设为0.6,可以过滤掉这些低置信度的类别,从而优化结果。修改后运行test.py,所有显示的类别置信度均高于60%。
摘要由CSDN通过智能技术生成

我在使用mask-rcnn分割图片的时候,出现了一张图片被识别为多中类别。但有些类别的置信度较低(<60%),所以采用以下方法抛出较低置信度的图片,以达到初步过滤的目的。

第一步:

在根目录下找到tools\test.py

第二步:

使用ctrl+f查找--show-score-thr

这个参数就是用来修改置信度的,我们将其设置为0.6,代表60%以下的数据将会被过滤。

测试:

修改为置信度后,再次运行test.py。

效果如下,数据置信度都>60%。

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 6
    评论
评论 6
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值