基于改进YOLOv7的路面病害检测算法OA北大核心CSTPCD
Pavement distress detection algorithm based on improved YOLOv7
针对公路路面病害图像存在光影变化大、背景干扰多、尺度差异大等问题,提出基于改进YOLOv7的路面病害检测算法.首先,对YOLOv7网络模型中的ELAN模块进行了优化,通过通道和空间注意力机制优化信息提取,增强网络对重要特征的提取能力;接着,使用ACmix注意力模块提高网络对小目标的关注度,有效解决原网络模型对小目标的漏检问题;其次,采用大下采样比率的卷积输出,提高对小目标的检测精度;最后,引入WIoUv3替换原网络模型中的CIoU来优化损失函数,构造梯度增益的计算方法来附加聚焦机制.实验结果表明:改进后的YOLOv7模型平均精度均值(mAP)与原模型相比提升了4.5%,检测效果优于原网络模型与传统经典目标检测网络模型.
In view of the substantial light and shadow variations,excessive background interference and large scale differences in highway pavement distress images,a pavement distress detection algorithm based on improved YOLOv7 is proposed.This research primarily focuses on the optimization of the ELAN module within the YOLOv7 network model,wherein an advanced amalgamation of channel and spatial attention mechanisms is employed to optimize the information extraction and elevate the network's ability in discerning and extracting important features.In order to mitigate the issue of missed detections of smaller objects in the original network model,the ACmix attention module is adeptly integrated,which amplifies the network's acuity towards smaller-scale objects.This paper introduces an innovative approach by adopting a convolutional outputs with heightened downsampling ratio to improve the precision in detecting smaller objects.The WIoUv3 is introduced to replace CIoU of the original network model to optimize the loss function,and the computation of gradient gain is constructed to attach the focusing mechanism.The experimental results show that the mean average precision(mAP)of the model based on improved YOLOv7 is increased by 4.5%relative to that of the original model,and its detection effect is better than not only the original network model but also the traditional classical object detection model.
葛焰;刘心中;马树森;赵津;李镇宏
福建理工大学 生态环境与城市建设学院,福建 福州 350118
电子信息工程
目标检测YOLOv7路面病害损失函数WIoUv3注意力机制
object detectionYOLOv7pavement distressloss functionWIoUv3attention mechanism
《现代电子技术》 2024 (011)
31-37 / 7
评论