首页|基于改进YOLOv7-tiny的雾天目标检测算法研究

基于改进YOLOv7-tiny的雾天目标检测算法研究

扫码查看
在户外场景下,目标检测是一项很重要的技术,其广泛应用于自动驾驶、监控目标跟踪等领域.然而户外天气多变,尤其在大雾天气场景下,由于光线减弱和物体边缘模糊等问题,导致算法性能下降、检测精度不高.针对这些问题,对雾天目标检测算法的研究提出了一种基于YOLOv7-tiny网络改进的算法.首先,引入注意力机制SimAM,在网络的特征融合部分添加注意力模块SimAM来提高网络对模糊图像的特征提取能力.然后,将算法原来的损失函数CIoU替换为wIoU,来提高锚框的定位精度,从而在整体上提高检测的精度.在雾天数据集ug2上进行了训练和验证.改进后的算法在计算量保持基本不变的前提下,mAP0.5和mAP0.5∶0.95分别提高了1.7和3.5个百分点,证明了改进方法的有效性.
Research on foggy target detection algorithm based on improved YOLOv7-tiny
In outdoor scenes,target detection is a very important technology,which is widely used in automatic driving,moni-toring target tracking and other fields.However,the outdoor weather is changeable,especially in the foggy weather scene,due to the problems of light weakening and object edge blur,the performance of the algorithm is degraded and the detection accuracy is not high.In order to solve these problems,a modified algorithm based on YOLOv7-tiny network was proposed in the study of target de-tection algorithm in foggy weather.Firstly,the attention mechanism SimAM was introduced,and the attention module SimAM was added to the feature fusion part of the network to improve the feature extraction ability of the network for blurred images.Then the original loss function CIoU of the algorithm was replaced by wIoU to improve the positioning accuracy of the anchor frame,thereby improving the accuracy of the detection as a whole.This paper has been trained and verified on the foggy dataset ug2.Under the premise that the calculation amount of the improved algorithm remains basically unchanged,mAP0.5 is increased by 1.7percentage,and mAP0.5:0.95 is increased by 3.5 percentage,which proves the effectiveness of the improved method.

target detectionfoggy weatherSimAMwIoU

高武阳、张麟华

展开 >

太原师范学院计算机科学与技术学院,晋中 030619

太原工业学院计算机工程系,太原 030008

目标检测 雾天 SimAM wIoU

2024

现代计算机
中大控股

现代计算机

影响因子:0.292
ISSN:1007-1423
年,卷(期):2024.30(5)
  • 18