[Object Detection] 1-Stage detector 와 2-Stage detector 란?

1-Stage Detector와 2-Stage Detector milestone

직선을 기준으로 위가 2-Stage Detector들이고 아래가 1-Stage Detector들이다.

 

 

2-Stage Detector

Regional Proposal과 Classification이 순차적으로 이루어진다.

 

Regional Proposal 이란?

기존에는 이미지에서 object detection을 위해 sliding window방식을 이용했었다.

Sliding window 방식은 이미지에서 모든 영역을 다양한 크기의 window (differenct scale & ratio)로 탐색하는 것이다.

 

<출처 : https://ssjin.tistory.com/entry/Computer-vision-Sliding-window-Detection >

 

 

이런 비효율성을 개선하기 위해 '물체가 있을만한' 영역을 빠르게 찾아내는 알고리즘이 region proposal이다.

대표적으로 Selective search, Edge boxes들이 있다.

 

(* Selective search : 비슷한 질감, 색, 강도를 갖는 인접 픽셀로 구성된 다양한 크기의 window를 생성한다.)

즉, regional proposal은 object의 위치를 찾는 localization문제이다.

 

 

따라서 2-stage detector에서 regional proposal과 classification이 순차적으로 이루어진다는 것은 classification과 localization문제를 순차적으로 해결한다는 것이다.

 

 

Selective search, Region proposal network와 같은 알고리즘을 및 네트워크를 통해 object가 있을만한 영역을 우선 뽑아낸다. 

이 영역을 RoI(Region of Interest)라고 하며, 이런 영역들을 우선 뽑아내고 나면 각 영역들을 convolution network를 통해 classification, box regression(localization)을 수행한다.

 

 

1-Stage Detector

2-stage detector와 반대로 regional proposal와 classification이 동시에 이루어진다.

즉, classification과 localization문제를 동시에 해결하는 방법이다. 

 

1-stage detector는 비교적 빠르지만 정확도가 낮고 2-stage detector는 비교적 느리지만 정확도가 높다.

 

 

RoI영역을 먼저 추출하지 않고 전체 image에 대해서 convolution network로 classification, box regression(localization)을 수행한다. 당연히 특정 object 하나만 담고있는 RoI에서 classification, localization을 수행하는 것보다 여러 noise 즉, 여러 object가 섞여있는 전체 image에서 이를 수행하는게 더 정확도는 떨어지지만 간단하고 쉬운만큼 속도가 빠르다는 장점이 있다.

 

 

 

📌 참고 사이트

< https://airsbigdata.tistory.com/211 >

< rhttps://ganghee-lee.tistory.com/34 >