Aiming at the problem that the high-performance face detection model has a large number of parameters,high computational complexity,and it is difficult to deploy on the edge of embedded devices,a lightweight face detection algorithm is proposed,which is based on the lightweight improvement on RetinaFace.Firstly,MobileNetV2_0.5× is used as the backbone to extract features,and PANLite with few parameters is selected for two-way fusion of multi-scale features to enhance the ability of feature representation.RFBLite is used as the feature enhancement module to incorporate context information while increasing the feature receptive field.Finally,filter pruning algorithm is applied in the trained model.Then the pruned model is retrained for fine-tuning and depolyed to Nvidia Jetson Nano for model inference.The experiment results show that the lightweight model can achieve high face detection performance with a small amount of parameters and low computational complexity,and can perform real-time inference on embedded platforms.
Key words
face detection /
filter pruning /
edge deployment /
Cortex-A57
{{custom_sec.title}}
{{custom_sec.title}}
{{custom_sec.content}}
References
[1] Kortli Y,Jridi M,Al Falou A,et al.Face recognition systems:A survey[J].Sensors,2020,20(2):342.
[2] Viola P A,Jones M J.Rapid Object Detection using a Boosted Cascade of Simple Features[C]//Computer Vision and Pattern Recognition,2001.CVPR 2001.Proceedings of the 2001 IEEE Computer Society Conference on.IEEE,2001.
[3] Deng J,Guo J,Ververas E,et al.Retinaface: Single-shot multi-level face localisation in the wild[C]//Proceedings of the IEEE/CVF conference on computer vision and pattern recognition,2020:5203-5212.
[4] He K,Zhang X,Ren S,et al.Deep residual learning for image recognition[C]//Proceedings of the IEEE conference on computer vision and pattern recognition,2016:770-778.
[5] Liu S,Huang D.Receptive field block net for accurate and fast object detection[C]//Proceedings of the European conference on computer vision (ECCV),2018:385-400.