计算机与现代化Issue(5):85-91,7.DOI:10.3969/j.issn.1006-2475.2024.05.015
基于改进YOLOv5的复杂路况密集行人检测方法
An Improved YOLOv5-based Method for Dense Pedestrian Detection Under Complex Road Conditions
摘要
Abstract
Aiming at the problem of low pedestrian detection accuracy in complex street scene environment,a new network YOLO-BEN is proposed based on the improvement of YOLOv5 network.The network uses a residual connection module Res2Net with hierarchical system to integrate with C3 module,enhancing fine-grained multi-scale feature representation.The paper adopts the Bi-level routing attention module to construct and prune a region level directed graph,and applies fine-grained atten-tion in the union of routing regions,enabling the network to have dynamic query aware sparsity and improving the feature extrac-tion ability of fuzzy images.We incorporate the EVC module to preserve local corner area information and compensate for the problem of information loss caused by occluded pedestrians.In this paper,NWD metric and original IoU metric are used to form a joint loss function,and a small target detection head is added to improve the effect of long-distance pedestrian detection.In the experiment,the method has achieved good results on self-made data sets and some WiderPerson data sets.Compared with the original network,the accuracy,recall and average accuracy of the improved network are increased by 2.8,4.3 and 3.9 percent-age points respectively.关键词
行人检测/多尺度特征/双层路由注意力机制/角区域特征/小目标检测Key words
pedestrian detection/multi scale features/double layer routing attention mechanism/angular areal feature/small target detection分类
信息技术与安全科学引用本文复制引用
孙睿琦,窦修超,李志华,蒋雪梅,孙宇豪..基于改进YOLOv5的复杂路况密集行人检测方法[J].计算机与现代化,2024,(5):85-91,7.基金项目
公安部科技强警基础工作计划项目(2022JC13) (2022JC13)