Non-max Suppression
前言
如何解決一個物件產生多個bounding boxes。
內容
Non-max suppression example
- 19x19個grid cell。
- 車輛的中心點理論上只會一個cell中,但實際上相臨近的grid cells都有機會抓到車輛的中心點。
- 因此就會生成多個bounding box。
- 如何決定出一個最相似的bounding box。
Non-max suppression algorithm
- 給每個bounding box一個機率值,選擇機率最高的。
- 具體作法:
- 先拋棄機率<0.6的bounding box。
- 選擇剩餘bounding box中機率最高的。
- 再拋棄剩餘bounding box中與選中bounding box的IOU> 0.5。
- 不斷重複,直到所有物品都只有一個bounding box。
- 對於多個class的識別,對於每一個class,都要獨立做一次Non-max suppression。
該學到的
- 理解non-max suppression 算法想要解決什麼問題。
- 理解non-max suppression 算法是如何運作的。