ObjectDetection - NMS

NMS ( Non - Maximum Suppression )

• 최종 박스 하나를 찾는 것

 

동작순서

1. 검출된 Bounding Box 중, Confidence_threshold보다 작으면 제거
2. Confidence Score를 기준으로 내림차순 정렬
3. 모든 박스 순차적으로 실행
   가장 높은 Confidence Score의 Bounding Box와 동일한 Class면서 
   iou_threshold이상인 Bounding Box는 제거
4. 남은 Bounding Box 선택

 

 

위와같이 class1(새)과 겹치는 부분이 많은데 같은 클래스라면 삭제한다.
즉, 가장 확률이 높은 박스 하나만 남기고 제거한다.

threshold가 너무 낮으면 조금만 겹쳐도 같은 클래스에 비슷한 위치라고 판단되어 박스가 모두 삭제되며 검출이 누락될 수 있고 너무 높으면 중복 박스가 많이 남아 결과가 지저분해질 수 있기 때문에 적절한 Iou_threshold값이 중요하다.