인공지능 객체인식

NMS (non-maximun-suppression)

JEO96 2022. 5. 15. 13:18
반응형

 

 

1. 서론

객체인식 모델 논문을 읽으면서 자주 쓰이는 용어 NMS를 다뤄보고자 한다.

2. NMS(non-maximun-suppression)

객체를 인식할 때 경우 아래 사진과 같이 한 객체를 여러 번 찾는 경우가 자주 생긴다. 같은 객체를 찾았다고 판단되는 가장 box 중에서 신뢰도(confidence)가 높은 box만 남기고 나머지 box를 제거하는 것을 NMS(비 최대 억제)라고 한다.

같은 객체라고 판단하는 방법은 IOU가 일정 수준을 넘긴 경우 넘긴 경우 같은 객체라고 판단하게 된다. 일반적으로 0.5를 넘긴 경우 같은 객체라고 하는 경우가 많다.

NMS를 실행하고 난 결과는 아래 사진과 같이 한 객체에 한 개의 box만 남게 된다.

 

반응형