多维度特征融合的双核卷积路面裂缝检测算法

Dual-kernel Convolutional Pavement Crack Detection Algorithm with Multi-dimensional Feature Fusion

  • 摘要: 对现有路面裂缝检测算法易受光照阴影、异物遮挡以及道路标线等条件影响,存在漏检误检、检测精度低、检测速度慢等问题进行了研究,提出了多维度特征融合的双核卷积路面裂缝检测算法。首先,添加三分支注意力机制,采用跨维度交互技术构建通道、高度和宽度维度间的相互依赖,赋予模型多维度特征提取能力;其次,使用双核卷积替换传统卷积实现模型的轻量化;再者,使用改进的MPDIoU边界框回归函数替换CIoU边界框回归函数,提高模型的收敛速度。在YOLOv8模型和RDD2022数据集上的实验表明,与YOLOv8n相比,F1和mAP分别提高了7.5%和5.3%,参数量和计算量分别下降了0.2%和0.7%,FPS提升了13;与SSD、Faster R-CNN、U-Net和DeepLabv3相比,F1提高了3.9%、mAP低于U-Net2%,参数量和计算量分别下降了0.8%和3.3%、FPS提升了19。

     

    Abstract: The existing pavement crack detection algorithms are easily affected by light and shadow, foreign object obstruction and road marking, which have the problems of missed detection, low detection accuracy and slow detection speed.Proposed a dual-kernel convolutional pavement crack detection algorithm with multi-dimensional feature fusion.Firstly, the triplet attention mechanism is added, cross-dimensional interaction technique is used to construct the interdependence between the channel, height and width dimensions, which endows the model with the ability of multi-dimensional feature fusion; secondly, the dual-kernel convolutional is used to replace the traditional convolutional to realize the model’s lightness; furthermore, the CIoU bounding box regression function is replaced by improved MPDIoU bounding box regression function, which improves the model’s convergence speed.Experiments on YOLOv8 model and RDD2022 dataset show that, compared with YOLOv8n, F1 and mAP are improved by 7.5% and 5.3%,the number of parameters and computation are decreased by 0.2% and 0.7%,respectively, the FPS is improved by 13;compared with SSD,Faster R-CNN,U-Net, and DeepLabv3,F1 is improved by 3.9%,mAP is 2% lower than U-Net, the number of parameters and computation decreased by 0.8% and 3.3% respectively, and the FPS increased by 19.

     

/

返回文章
返回