我在使用mask-rcnn分割图片的时候,出现了一张图片被识别为多中类别。但有些类别的置信度较低(<60%),所以采用以下方法抛出较低置信度的图片,以达到初步过滤的目的。
![](https://i-blog.csdnimg.cn/blog_migrate/6f91cbbb685cc0824b2548c957330152.png)
第一步:
在根目录下找到tools\test.py
![](https://i-blog.csdnimg.cn/blog_migrate/7d0227a46e20fdaeac8626b6d1d83bed.png)
第二步:
使用ctrl+f查找--show-score-thr
这个参数就是用来修改置信度的,我们将其设置为0.6,代表60%以下的数据将会被过滤。
![](https://i-blog.csdnimg.cn/blog_migrate/55cd826bfdbc0f08ef6ae340945c5f4a.png)
测试:
修改为置信度后,再次运行test.py。
效果如下,数据置信度都>60%。
![](https://i-blog.csdnimg.cn/blog_migrate/40427515b9601f87102cb7fb5c42ddd7.png)
我在使用mask-rcnn分割图片的时候,出现了一张图片被识别为多中类别。但有些类别的置信度较低(<60%),所以采用以下方法抛出较低置信度的图片,以达到初步过滤的目的。
第一步:
在根目录下找到tools\test.py
第二步:
使用ctrl+f查找--show-score-thr
这个参数就是用来修改置信度的,我们将其设置为0.6,代表60%以下的数据将会被过滤。
测试:
修改为置信度后,再次运行test.py。
效果如下,数据置信度都>60%。