1. Mobile Centernet for Embedded Deep Learning Object Detection
- Author
-
Chang Wen Chen, Mengyan Li, Jun Yu, Ye Yu, Haonian Xie, and Guochen Xie
- Subjects
Artificial neural network ,Computer science ,business.industry ,Deep learning ,05 social sciences ,Mobile computing ,Inference ,010501 environmental sciences ,01 natural sciences ,Object detection ,Task (project management) ,Computer engineering ,0502 economics and business ,Key (cryptography) ,Artificial intelligence ,050207 economics ,business ,0105 earth and related environmental sciences ,Data compression - Abstract
Object detection is a fundamental task in computer vision with wide application prospect. And recent years, many novel methods are proposed to tackle this task. However, most algorithms suffer from high computation cost and long inference time, which makes them impossible to be deployed on embedded devices in real industrial application scenarios. In this paper, we propose the Mobile CenterNet to solve this problem. Our method is based on CenterNet but with some key improvements. To enhance detection performance, we adopt HRNet as a powerful backbone and introduce a categorybalanced focal loss to deal with category imbalance problem. Moreover, to compress the model size as well as reduce inference time, knowledge distillation is employed to transfer knowledge from cumbersome model to a compact one. We conduct experiments on a large traffic detection dataset BDD100K and validate the effectiveness of all the modifications. Finally, our method achieves the 1st place in the Embedded Deep Learning Object Detection Model Compression Competition held in ICME 2020.
- Published
- 2020
- Full Text
- View/download PDF