作者投稿和查稿 主编审稿 专家审稿 编委审稿 远程编辑

计算机工程 ›› 2019, Vol. 45 ›› Issue (7): 203-207. doi: 10.19678/j.issn.1000-3428.0050951

• 人工智能及识别技术 • 上一篇    下一篇

基于特征金字塔网络的人群计数算法

马皓, 殷保群, 彭思凡   

  1. 中国科学技术大学 自动化系, 合肥 230026
  • 收稿日期:2018-03-27 修回日期:2018-05-24 出版日期:2019-07-15 发布日期:2019-07-23
  • 作者简介:马皓(1993-),男,硕士研究生,主研方向为人群计数算法;殷保群,教授;彭思凡,硕士研究生。
  • 基金资助:
    总装预研基金(61403120201)。

Crowd Counting Algorithm Based on Feature Pyramid Network

MA Hao, YIN Baoqun, PENG Sifan   

  1. Department of Automation, University of Science and Technology of China, Hefei 230026, China
  • Received:2018-03-27 Revised:2018-05-24 Online:2019-07-15 Published:2019-07-23

摘要: 由于单张图片人群计数存在严重的人群遮挡和尺度变化问题,导致人群计数算法性能明显下降。为此,提出一种基于特征金字塔网络对图片进行人群计数的算法,并给出能够处理任意图片分辨率的全卷积网络。将特征金字塔网络应用到人群计数中,通过逐层融合网络中不同尺度的特征图来解决图片中的上述问题。在人群计数数据库ShanghaiTech上对网络模型进行训练和性能评测,结果表明,与当前主流的人群计数算法相比,该算法具有更高的鲁棒性和准确性。

关键词: 人群计数, 卷积神经网络, 特征金字塔, 密度图, 平均绝对误差

Abstract: The single-picture crowd count has a sharp decline in performance due to severe population occlusion and scale changes.Therefore,this paper proposes an algorithm for crowd counting pictures,and gives a Full Convolution Network(FCN) capable of processing the resolution of any picture.The scale change and occlusion problems in the picture are solved by applying the feature pyramid network to the crowd count.The network model is trained and evaluated in the crowd counting database ShanghaiTech,results show that the algorithm has good robustness and accuracy compared with the current mainstream crowd counting algorithm.

Key words: crowd counting, Convolutional Neural Network(CNN), feature pyramid, density map, Mean Absolute Error(MAE)

中图分类号: