基于Yolov5的密集场所人数估计方法

作者：王婧媛; 方健^*

来源：吉林大学学报(信息科学版), 2021, 39(06): 682-687.

DOI：10.19292/j.cnki.jdxxp.2021.06.008

• 人群计数
• Mosaic数据增强
• Yolov5算法
• Focus和CSP结构

为解决目前对高密度人群计数问题,提出了一种基于Yolov5的人群计数方法。其中输入层主要进行Mosaic数据增强,即自适应锚框和自适应的图片缩放技术;Backbone中Yolov5主要采用Focus和CSP(Cross Stage Partial)结构;Neck层采用SPP(Spatial Pyramid Pooling)模块和FPN(Feature Pyramid Networks)+PAN(Pixel Aggregation Network)结构;输出端主要针对Bounding Box损失函数采用了CIOULoss作为损失函数和DIOUloss作为NMS(Non Maximum Suppression)的平均指标;最终输出训练结果。实验结果表明,该方法能有效提高人群计数的精度。

• 单位
  吉林工程技术师范学院