[Boostcamp AI Tech] [P-stage] 10주차 Day 1 학습 기록

bluegun·2021년 10월 6일
0

AI Tech 2기 활동

목록 보기
42/87

학습 내용

Advanced Object detection - CascadeRCNN, DCN, Swin


Cascade R-CNN

Faster R-CNN에서는 RPN을 학습할 때 IoU > 0.7인 Anchor box를 positive sample, IoU < 0.3인 Anchor box를 negative sample로 정했다. 또 Head를 학습하기 위해 IoU > 0.5인 RoI를 positive sample, IoU < 0.5인 RoI를 negative sample로 정했다. 여기서 0.5는 연구자들이 임의로 정한 수치로, cascade R-CNN은 IoU 값을 어떤 기준으로 나눠야 효율적일지에 대해 집중하였다.

연구를 통해 IoU threshold에 따라 결과가 다르고, 특히 Input IoU가 높을수록 높은 IoU threshold에서 학습된 model이 성능이 좋았다는 결론을 얻게 되었다.

0개의 댓글