DOI QR코드

DOI QR Code

Objedet detection using TensorRT engine and SSD

TensorRT 엔진과 SSD를 이용한 Face detection

  • Yoo, Hye-Bin (Dept of Electrical, Electronic and Control, Hankyong National University) ;
  • Kim, Sang-Hoon (Dept of Electrical, Electronic and Control, Hankyong National University)
  • 유혜빈 (한경대학교 전기전자제어공학과) ;
  • 김상훈 (한경대학교 전기전자제어공학과)
  • Published : 2020.05.29

Abstract

최근에는 딥러닝 기술의 발달로 물체 인식 및 검출에 관한 기술들 또한 발탄하고 있다. 검출에 관한 여러 기법(Faster R-CNN, R-CNN, YOLO, SSD 등) 중 SSD는 다른 기법들과는 다르게 높은 정확도와 빠른 속도가 특징이다. 동시에 여러 detection network들도 쉽게 이용이 가능하다. 본 논문에서는 detection netowork중 Mobilenet V2 network를 이용하여 SSD와 결합해 모델을 훈련하고, TensorRT engine을 이용하여 더 빠른 속도로 검출할 수 있는 방법에 대해 논의한다. 이 방법을 통해 face detector를 만들어 여러 상황에서 쓰일 수 있도록 한다.

Keywords