计算机工程2024，Vol.50Issue(3)：250-258,9.DOI:10.19678/j.issn.1000-3428.0067741
面向拥挤行人检测的改进YOLOv7算法
Improved YOLOv7 Algorithm for Crowded Pedestrian Detection
摘要
Abstract
Aiming at the problem that the detection algorithm is prone to omission and false detection in crowded pedestrian detection scenarios,this study proposes an improved YOLOv7 crowded pedestrian detection algorithm.Introducing a BiFormer visual transformer and an improved RepConv and Channel Space Attention Module(CSAM)-based Efficient Layer Aggregation Network(RC-ELAN)module in the backbone network,the self-attention mechanism and the attention module enable the backbone network to focus more on the important features of the occluded pedestrians,effectively mitigating the adverse effects of the missing target features on the detection.The improved neck network based on the idea of a Bidirectional Feature Pyramid Network(BiFPN)is used,and the transposed convolution and improved Rep-ELAN-W module enable the model to efficiently utilize the small-target feature information in the middle and low-dimensional feature maps,effectively improving the small-target pedestrian detection performance of the model.The introduction of an Efficient Complete Intersection-over-Union(E-CIoU)loss function allows the model to further converge to a higher accuracy.Experimental results on the WiderPerson dataset containing a large number of small target-obscuring pedestrians demonstrate that the average accuracies of the improved YOLOv7 algorithm when the IoU thresholds are set to 0.5 and 0.5-0.95 are improved by 2.5 and 2.8,9.9 and 7.1,and 12.3 and 10.7 percentage points compared with the YOLOv7,YOLOv5,and YOLOX algorithms,respectively,which can be better applied to crowded pedestrian detection scenarios.关键词
机器视觉/拥挤行人检测/注意力机制/YOLO系列算法/双向特征金字塔网络Key words
machine vision/crowded pedestrian detection/attention mechanism/YOLO series algorithms/Bi-directional Feature Pyramid Network(BiFPN)分类
信息技术与安全科学引用本文复制引用
徐芳芯,樊嵘,马小陆..面向拥挤行人检测的改进YOLOv7算法[J].计算机工程,2024,50(3):250-258,9.基金项目
国家自然科学基金(62172004,61872004) （62172004,61872004）
安徽省科技重大专项(202003a05020028) （202003a05020028）
安徽省高等学校自然科学研究重点项目(KJ2019A0065) （KJ2019A0065）
芜湖市核心技术攻关科技计划项目(2022hg10). （2022hg10）
