计算机应用 ›› 2019, Vol. 39 ›› Issue (8): 2217-2222.DOI: 10.11772/j.issn.1001-9081.2019010164

• 人工智能 • 上一篇    下一篇

基于深度学习的ARM平台实时人脸识别

方国康1,2, 李俊1,2, 王垚儒1,2   

  1. 1. 武汉科技大学 计算机科学与技术学院, 武汉 430065;
    2. 智能信息处理与实时工业系统湖北省重点实验室(武汉科技大学), 武汉 430065
  • 收稿日期:2019-01-22 修回日期:2019-03-14 出版日期:2019-08-10 发布日期:2019-04-15
  • 通讯作者: 李俊
  • 作者简介:方国康(1994-),男,湖北恩施人,硕士研究生,主要研究方向:智能计算、计算机视觉;李俊(1978-),男,湖北黄石人,副教授,博士,主要研究方向:智能计算、机器学习;王垚儒(1995-);男,湖北天门人,硕士研究生,主要研究方向:智能计算、深度学习。
  • 基金资助:
    国家自然科学基金资助项目(61572381);武汉科技大学智能信息处理与实时工业系统湖北重点实验室基金资助项目(znxx2018QN06)。

Real-time face recognition on ARM platform based on deep learning

FANG Guokang1,2, LI Jun1,2, WANG Yaoru1,2   

  1. 1. School of Computer Science and Technology, Wuhan University of Science and Technology, Wuhan Hubei 430065, China;
    2. Hubei Province Key Laboratory of Intelligent Information Processing and Real-time Industrial System(Wuhan University of Science and Technology), Wuhan Hubei 430065, China
  • Received:2019-01-22 Revised:2019-03-14 Online:2019-08-10 Published:2019-04-15
  • Supported by:
    This work is partially supported by the National Natural Science Foundation of China (61572381), the Hubei Key Laboratory of Intelligent Information Processing and Real-time Industrial System Foundation of Wuhan University of Science and Technology (znxx2018QN06).

摘要: 针对ARM平台上人脸识别实时性不强和识别率低的问题,提出一种基于深度学习的实时人脸识别方法。首先基于MTCNN人脸检测算法设计了一种实时检测并追踪人脸的算法;然后在ARM平台上基于深度残差网络(ResNet)设计人脸特征提取网络;最后针对ARM平台的特点,使用Mali-GPU加速人脸特征提取网络的运算,分担CPU负荷,提高系统整体运行效率。算法部署在基于ARM的瑞芯微RK3399开发板上,运行速度达到22 帧/s。实验结果表明,与MobileFaceNet相比,该方法在MegaFace上的识别率提升了11个百分点。

关键词: ARM平台, 人脸识别, 人脸追踪, 残差网络, Mali-GPU

Abstract: Aiming at the problem of low real-time performance of face recognition and low face recognition rate on ARM platform, a real-time face recognition method based on deep learning was proposed. Firstly, an algorithm for detecting and tracking faces in real time was designed based on MTCNN face detection algorithm. Then, a face feature extraction network was designed based on Residual Neural Network (ResNet) on ARM platform. Finally, according to the characteristics of ARM platform, Mali-GPU was used to accelerate the operation of face feature extraction network, sharing the CPU load and improving the overall running efficiency of the system. The algorithm was deployed on ARM-based Rockchip development board, and the running speed reaches 22 frames per second. Experimental results show that the recognition rate of this method is 11 percentage points higher than that of MobileFaceNet on MegaFace.

Key words: ARM platform, face recognition, face tracking, Residual neural Network (ResNet), Mali-GPU

中图分类号: