NMS : 여러 Bounding Box 중 확실한 Box만 선택하는 기법

NMS : 여러 Bounding Box 중 확실한 Box만 선택하는 기법

여러 개의 Bounding Box를 가지는 이유 : selective search 경우에도 object가 있을만한 공간을 box를 침. object가 있는데 실수를 하면 안 되기 때문

NMS 진행 과정

1. Detected 된 bounding box 별로 특정 Confidence threshold 이하 bounding box는 가장 먼저 제거
2. 가장 높은 confidence score를 가진 box 순으로 내림차순 정렬하고 아래 로직을 모든 box에 순차적으로 적용.
   • 정렬된 confidence score를 가진 box를 차례로 기준으로 삼고 다른 box들과 IOU를 구함.
   • 구한 IOU가 특정 threshold 이상인 box를 모두 제거.
3. 남아 있는 box만 선택

즉 Confidence score가 높을 수록, IOU threshold가 낮을 수록 많은 box가 제거된다.

 

위 그림을 예시로 들어보겠다.

위 그림은 1번 과정이 진행됐다고 가정한다.

1. 내림차순으로 정렬을 하면 [98, 96, 90, 87, 82, 82] 일 것이다.
2. 98% 박스와 나머지[96, 90, 87, 82, 82] 박스들과 IOU를 계산한다. if IOU threshold = 0.4
3. 98과 96는 겹치지 않아 IOU가 0일 것이다. 따라서 96은 생존한다.
4. 98과 [90, 82(왼 노랑)]의 경우 겹치는 정도가 크기 때문에 IOU가 0.4보다 크다고 가정한다면 삭제 대상이다.
5. 다음 96을 기준으로 [87, 82(오른 노랑)]의 경우 마찬가지로 크게 겹쳐 IOU가 크고 삭제 대상이다.
6. 이처럼 98, 96제외 모든 박스가 제거되면서 확실한 박스를 남기는 것이 NMS이다.