Lightweight SSD: Real-time Lightweight Single Shot Detector for Mobile Devices

Shi Guo, Yang Liu, Yong Ni, Wei Ni

2021

Abstract

Computer vision has a wide range of applications, and the current demand for intelligent embedded terminals is increasing. However, most research on CNN (Convolutional Neural Network) detectors did not consider mobile devices' limited computation and did not specifically design networks for mobile devices. To achieve an efficient object detector for mobile devices, we propose a lightweight detector named Lightweight SSD. In the backbone part, we design our MBlitenet backbone based on the Attentive linear inverted residual bottleneck to enhance the backbone's feature extraction capability while achieving the lightweight requirements. In the detection neck part, we propose an efficient feature fusion network CFPN. Two innovative and useful Bag of freebies named BLL loss (Both Localization Loss) and GrayMixRGB are applied to the Lightweight SSD’s training procedure. They can further improve detector capabilities and efficiency without increasing the inference computation. As a result, Lightweight SSD achieves 74.4 mAP (mean Average Precision) with only 4.86M parameters on PASCAL VOC, being 0.2x smaller yet still more accurate 3.5 mAP than the previous best lightweight detector. To our knowledge, the Lightweight SSD is the state-of-the-art real-time lightweight detector on mobile devices with the edge Application-specific integrated circuit (ASIC). Source Code will be released after paper publication.

Paper Citation

in Harvard Style

Guo S., Liu Y., Ni Y. and Ni W. (2021). Lightweight SSD: Real-time Lightweight Single Shot Detector for Mobile Devices. In Proceedings of the 16th International Joint Conference on Computer Vision, Imaging and Computer Graphics Theory and Applications (VISIGRAPP 2021) - Volume 5: VISAPP; ISBN 978-989-758-488-6, SciTePress, pages 25-35. DOI: 10.5220/0010188000250035

in Bibtex Style

@conference{visapp21,
author={Shi Guo and Yang Liu and Yong Ni and Wei Ni},
title={Lightweight SSD: Real-time Lightweight Single Shot Detector for Mobile Devices},
booktitle={Proceedings of the 16th International Joint Conference on Computer Vision, Imaging and Computer Graphics Theory and Applications (VISIGRAPP 2021) - Volume 5: VISAPP},
year={2021},
pages={25-35},
publisher={SciTePress},
organization={INSTICC},
doi={10.5220/0010188000250035},
isbn={978-989-758-488-6},
}

in EndNote Style

TY - CONF

JO - Proceedings of the 16th International Joint Conference on Computer Vision, Imaging and Computer Graphics Theory and Applications (VISIGRAPP 2021) - Volume 5: VISAPP
TI - Lightweight SSD: Real-time Lightweight Single Shot Detector for Mobile Devices
SN - 978-989-758-488-6
AU - Guo S.
AU - Liu Y.
AU - Ni Y.
AU - Ni W.
PY - 2021
SP - 25
EP - 35
DO - 10.5220/0010188000250035
PB - SciTePress