A Study of Object Detection on Vision-transformer-based One-stage Detector for Threat Inspection of Passenger Baggages
Ji-Wook Jeong, Yoonseon Song, Sooyeul Lee · Journal of the Institute of Electronics and Information Engineers · 2022
본 논문에서는, 휴대 수하물 x-선 영상 기반 위험화물 검출을 위한 실시간 육안 검사를 보완하기 위한 기초연구를 위해 비전 트랜스포머 기반 One-stage 디텍터를 이용한 x-ray 영상 기반 데이터셋 객체 검출 연구를 수행하고자 하였다. 최근 뛰어난 객체 검출 성능을 보인 바 있는 PAA (Probabilistic Anchor Assignment) 디텍터에 다양한 디텍터, 주의 함수 알고리즘을 적용하여 검출 성능 개선을 위한 실험을 수행한 결과, Swin Transformer 백본을 이용한 PAA 디텍터에 FreeAnchor 알고리즘을 부분 적용하고, Dynamic Head 주의 함수 모듈을 추가 결합하여 학습을 진행한 결과, SIXray-10 데이터셋에 대해 64.0 mAP의 검출 성능을 보였다.