DOI QR코드

DOI QR Code

저전력 장치를 위한 자원 효율적 객체 검출기

Resource-Efficient Object Detector for Low-Power Devices

  • 투고 : 2023.09.21
  • 심사 : 2024.01.12
  • 발행 : 2024.01.31

초록

본 논문은 전통적인 자원 집약적인 컴퓨터 비전 모델의 한계를 해결하기 위해 저전력 엣지 장치에 최적화된 새로운 경량 객체 검출 모델을 제안합니다. 제안된 검출기는 Single Shot Detector (SSD)에 기반하여 소형이면서도 견고한 네트워크를 설계하였고, 작은 객체를 효율적으로 감지하는 데 있어 효율성을 크게 향상시키도록 모델을 구성하였다. 이 모델은 주로 두 가지 구성요소로 구성되어 있습니다: Depthwise 와 Pointwise Convolution 레이어를 사용하여 효율적인 특징 추출을 위한 Light_Block, 그리고 작은 객체의 향상된 감지를 위한 Enhancer_Block 으로 나누었다. 우리의 모델은 300x480 의 이미지 크기를 가진 Udacity 주석이 달린 데이터셋에서 처음부터 훈련되었으며, 사전 훈련된 분류 가중치의 필요성을 제거하였다. 약 0.43M 의 파라미터로 5.5MB 만의 무게를 가진 우리의 검출기는 평균 정밀도 (mAP) 27.7%와 140 FPS 의 처리 속도를 달성하여, 정밀도와 효율성 모두에서 기존 모델을 능가하였다. 따라서, 본 논문은 추론의 정확성을 손상시키지 않으면서 엣지 장치를 위한 객체 검출에서의 효과적인 경량화를 보여주고 있다.

This paper presents a novel lightweight object detection model tailored for low-powered edge devices, addressing the limitations of traditional resource-intensive computer vision models. Our proposed detector, inspired by the Single Shot Detector (SSD), employs a compact yet robust network design. Crucially, it integrates an 'enhancer block' that significantly boosts its efficiency in detecting smaller objects. The model comprises two primary components: the Light_Block for efficient feature extraction using Depth-wise and Pointwise Convolution layers, and the Enhancer_Block for enhanced detection of tiny objects. Trained from scratch on the Udacity Annotated Dataset with image dimensions of 300x480, our model eschews the need for pre-trained classification weights. Weighing only 5.5MB with approximately 0.43M parameters, our detector achieved a mean average precision (mAP) of 27.7% and processed at 140 FPS, outperforming conventional models in both precision and efficiency. This research underscores the potential of lightweight designs in advancing object detection for edge devices without compromising accuracy.

키워드

과제정보

This research was supported by Basic Science Research Program through the National Research Foundation of Korea(NRF) funded by the Ministry of Education (NRF-2021R1F1A1064351).

참고문헌

  1. X. Xu, X. Zhang, T. Zhang, "Lite-YOLOv5: A Lightweight Deep Learning Detector for On-Board Ship Detection in Large-Scene Sentinel-1 SAR Images", Remote Sensing, vol. 14, no. 4, pp. 1-27, Feb. 2022. https://doi.org/10.3390/rs14041018
  2. L. Yang, G. Yuan, H. Wu, W. Qian, "An ultra-lightweight detector with high accuracy and speed for aerial images", Math Biosci Eng. vol. 20, no. 8, pp. 13947-13973, Jun. 2023. https://doi.org/10.3934/mbe.2023621
  3. W.Liu, D.Anguelov, D. Erhan, C.Szegedy, S. Reed, C.Fu, A.C. Berg, "SSD: Single Shot MultiBox Detector," European Conference on Computer Vision(ECCV), pp. 21-37, Oct. 2015.
  4. A. G. Howard, M. Zhu, B. Chen, D. Kalenichenko, W. Wang, T. Weyand, M. Andreetto, H. Adam, "Mobilenets: Efficient convolutional neural networks for mobile vision applications," CVPR, April 2017.
  5. [online]Available:https://github.com/pierluigiferrari/ssd_keras/blob/master/ssd7_training.ipynb [Accessed on SEPT. 5, 2023].
  6. Y. Yamashige and M. Aono, "FPSSD7: Real-time Object Detection using 7 Layers of Convolution based on SSD," International Conference of Advanced Informatics: Concepts, Theory, and Applications (ICAICTA), pp. 1-6, Sept. 2019.