一种基于DRB轻量化模块和轻量化检测头的YOLOv8网络的路面缺陷检测方法

    公开(公告)号:CN119599933A

    公开(公告)日:2025-03-11

    申请号:CN202411081150.X

    申请日:2024-08-08

    Abstract: 本发明提供一种基于DRB轻量化模块和轻量化检测头的YOLOv8网络的路面缺陷检测方法,首先使用DRB轻量化模块融合到C2f模块中,替换C2f中Bottleneck残差模块中的卷积结构,构成C2f_DRB模块来组成YOLOv8的backbone网络,使backbone网络结构可以更好的提取特征信息,同时通过冲参数化,扩大卷积感受野,适应不同尺度特征;然后对YOLOv8的检测头进行改进,把原来参数量和计算量较高的头部中的CBS模块进行改进,选择参数量和计算量更低的组卷积和深度可分离卷积对CBS替换。在RDD2020路面缺陷数据集上进行实验,YOLOv8模型相较于原始的YOLOv8模型,提出改进后的网络C2f_DRB结构,网络的参数量减少了8.1%,计算量减少了8.4%,准确率提升了2.1%,改进后的检测头的网络的参数量减少了14.1%,计算量减少了24%。改进的模型整体有了一个较为明显的提升效果。结果表明,本发明提出改进的模型有效的减少了参数量和计算量,以及在准确性方面也有提高。

Patent Agency Ranking