关键词:
目标检测
密集人群
YOLOv5
注意力机制
摘要:
在地震、火灾、恐怖袭击等事件发生后,需要通过目标检测算法实时协助进行人员疏散和监管。但是由于人群姿态复杂,同时存在遮挡和密集分布等因素,算法精确度不足。因此本文提出一种改进的密集人群目标检测算法,在YOLOv5的基础上分别对骨干网和Neck进行改进,引入了CBAM(Convolutional Block Attention Module)注意力机制和双向特征金字塔网络(Bidirectional Feature Pyramid Network,BiFPN),从而提升算法对特征信息提取能力,提高目标检测的准确性。最终通过与主流算法对比,证明改进后的算法参数量仅为7.12M、计算量为16.1GFLOPs,mAP@50高达41.7%。通过消融实验,证明在整体类别上精确率和mAP@50提高了17.4%、2.4%,密集人群的精确率达到41.2%,均达到了最优,为应急救援和管理提供了理论依据和数据支持。