1. 程式人生 > >目標檢測 — NMS

目標檢測 — NMS

spa 置信度 ssi 檢測 pan 最大的 我們 效果 保留

1、非極大值抑制步驟

  非極大值抑制算法(Non-maximum suppression,NMS)在目標檢測中經常用到。我們的檢測算法可能對同一目標產生多次檢測的結果,非極大值抑制算法可以保證每個目標只檢測一次,找到檢測效果最好的框。

(1)去除所有預測框置信度於某個閾值的框,這裏的閾值選取為0.5。

(2)在剩下的1、2、4、5、6號框中選取置信度最大的,即圖中5號框,比較除5號框以外所有框於5號框的交並比,若交並比大於某個閾值(人工給定的值),則抑制這些不是局部極大值預測框輸出。

(3)再從剩下的框中選取置信度最大的,重復上面的步驟,直至沒有框被舍去時結束。

(4)最後2號和5號框被保留,其他框被抑制。

  若檢測物體是多個類別的時候,一般情況是對每個類別分別做一次非極大值抑制算法。

目標檢測 — NMS