基于改进YOLOv7-tiny的户外行人检测算法研究
Research on outdoor pedestrian detection algorithm based on improved YOLOv7-tiny
潘兴好 1郭彩萍2
作者信息
- 1. 太原师范学院计算机科学与技术学院,晋中 030619
- 2. 太原工业学院电子工程系,太原 030008
- 折叠
摘要
在行人检测的高密度交通场景中,检测算法通常会漏掉被遮挡和远处的模糊行人,同时无法兼顾检测的精度和速度.针对这些问题,基于YOLOv7-tiny提出一种改进的户外行人检测算法.该算法引入SENet注意力机制,抑制了不相关的信息,以此提高特征图表达信息的能力,同时增强了对行人目标特征的提取.为了更好地识别目标的边缘和重叠情况,提高回归精度,在损失函数中用SIoU代替CIoU,提升了遮挡情况下的检测率.根据在WiderPerson数据集上的实验,在保证检测速度的前提下,对比YOLOv7-tiny检测算法,平均检测精度提升了2个百分点.实验结果表明,经过改进的算法可以显著提高检测性能.
Abstract
In high-density traffic scenarios for pedestrian detection,detection algorithms usually miss obscured and distant fuzzy pedestrians,while failing to balance detection accuracy and speed.To address these problems,an improved outdoor pedes-trian detection algorithm based on YOLOv7-tiny is proposed.The algorithm introduces the SENet attention mechanism,which sup-presses irrelevant information as a way to improve the ability of the feature map to express information,and at the same time en-hances the extraction of pedestrian target features.In order to better recognize the edges and overlapping of targets and improve the regression accuracy,SIoU is used instead of CIoU to improve the detection rate in the case of occlusion.According to the experi-ments on the WiderPerson dataset,the average detection accuracy is improved by 2 percentage comparing with the YOLOv7-tiny detection algorithm under the premise of guaranteeing the detection speed.The experimental results show that the improved algo-rithm can significantly improve the detection performance.
关键词
行人检测/YOLOv7-tiny/SENet/SIoUKey words
pedestrian detection/YOLOv7-tiny/SENet/SIoU引用本文复制引用
出版年
2024
NETL
NSTL
万方数据