본문 바로가기
카테고리 없음

NMS( non-maximum-suppression)

by Wanda 2022. 2. 20.

*비 최대 억제 (non maximum suppression) 

 : computer vision tasks에서 많은 overlapping 된 entities들 중 하나의 entity를 선택하기 위한 알고리즘 

 

과정

 

step1) 가장 높은 confidence score를 가지고 있는 prediction S를 선택 한 후 P라는 목록에서 제거한 후 이것을 final prediction 리스트(keep)에 더한다. 

 

2) 그리고 keep에 있는  prediction SP에 있는 모든 predictions과 비교한다(IOU 이용). 만약 IOU가 thresh_iou 보다 높은 값들이 있으면, P에서 제거한다. 

 

step 3) P에서 여전히 남아있는 예측 값이 있으면, step1으로 돌아가서 다시 진행하고, 그렇지 않으

면 keep이라는 list에 돌아간다.