计算机工程与应用 ›› 2019, Vol. 55 ›› Issue (12): 162-168.DOI: 10.3778/j.issn.1002-8331.1805-0441

• 模式识别与人工智能 • 上一篇    下一篇

基于深度学习的实时场景小脸检测方法

叶  锋,赵兴文,宫恩来,杭丽君   

  1. 杭州电子科技大学 自动化学院,杭州 310018
  • 出版日期:2019-06-15 发布日期:2019-06-13

Tiny Face Detection Based on Deep Learning Inreal-Time Scenes

YE Feng, ZHAO Xingwen, GONG Enlai, HANG Lijun   

  1. College of Automation, Hangzhou Dianzi University, Hangzhou 310018, China
  • Online:2019-06-15 Published:2019-06-13

摘要: 实时场景下的小脸检测存在检出率低而且回归精度差的问题。通过融合更底层特征进行多尺度级联预测。根据实时场景下的人脸特点生成不同大小和比例的预测框以更好地适应人脸形状。在预测阶段提出了基于IOU判别的soft and hard nms算法,对冗余预测框进行抑制,设置两个阈值将网络生成的预测框划分为低中高三段,对不同段的预测框采取不同的处理以达到精准筛选的目的。最优架构可在两张NVIDIA GTX 1080显卡下的实时视频检测和摄像头检测中获得45 f/s的速度,并且在Wider Face总体验证集上取得82.6%的平均精度。

关键词: 深度学习, 小脸检测, 实时检测, 计算机视觉

Abstract: Tiny face detection in real-time scenes has a low detection rate and poor regression accuracy. This paper further integrates the lower-level feature maps for multi-scale prediction. According to the characteristics of face in real-time scene detection, predicted boxes of different scales are generated to better adapt to human face shape. In the prediction stage, a soft and hard nms algorithm based on Intersection of Union(IOU) discrimination is proposed to suppress the redundant prediction boxes. Two thresholds are set to divide the prediction frame generated by the network into three segments of low, medium and high, and different segments of the prediction boxes are treated differently to achieve accurate suppress. The optimal architecture of the paper can obtain 45 frame per second in real-time video detection and camera detection under two NVIDIA GTX 1080 graphics cards, and achieves an average accuracy of 82.6% on the Wider Face overall validation set.

Key words: deep learning, tiny face detection, real-time detection, computer vision