融合可变形卷积的轻量级路面病害检测算法
DOI:
作者:
作者单位:

桂林电子科技大学信息与通信学院

作者简介:

通讯作者:

中图分类号:

TP391.4

基金项目:

广西信息材料重点实验室基金(221035-K);中央引导地方科技发展资金项目(20231011);广西高校中青年教师基础能力提升项目(2022KY0195)


A lightweight road disease detection algorithm fused with deformable convolution
Author:
Affiliation:

School of information and communication, Guilin University of Electronic technology

Fund Project:

  • 摘要
  • |
  • 图/表
  • |
  • 访问统计
  • |
  • 参考文献
  • |
  • 相似文献
  • |
  • 引证文献
  • |
  • 资源附件
  • |
  • 文章评论
    摘要:

    针对现有的路面病害检测算法在复杂环境下检测精度低,模型复杂度高的问题,在YOLOv5基础上,提出一种融合可变形卷积的轻量级路面病害检测算法(lightweight deformable convolution YOLOv5,LDC-YOLOv5)。首先,针对真实路面病害复杂不规整的特点,使用可变形卷积(Deformable Conv)和深度卷积(Depthwise Conv),设计了一种轻量级特征提取模块,代替原网络主干部分的C3模块,使卷积核聚焦在无规则裂缝病害上,增强病害特征提取能力。其次,针对特征融合阶段出现算法复杂度过高的问题,使用轻量级卷积GhostConv,构建一种轻量级特征融合模块,代替原网络颈部网络部分的C3模块,降低网络参数和复杂度;为避免真实路面出现光照不均,出现阴影遮挡路面病害目标而造成的病害漏检的情况,在主干网络部分,引入轻量级注意力机制TripletAttention,增强算法对病害信息上下文之间的理解能力。最后在IEEE公开数据集RDD2022和Kaggle公开数据集Road Damage上进行测试,实验结果表明,与YOLOv5s相比,mAP50在两个数据集上分别提升了1.4%和4.2%,且模型参数量仅为YOLOv5s的67.6%。

    Abstract:

    In response to the low detection accuracy and high model complexity issues of existing road damage detection algorithms in complex environments, a lightweight road damage detection algorithm, named Lightweight Deformable Convolution YOLOv5 (LDC-YOLOv5), is proposed based on YOLOv5. Firstly, to address the complexity of real road surface damages, a lightweight feature extraction module is designed using deformable convolution (Deformable Conv) and depthwise convolution (Depthwise Conv) to replace the C3 module in the original network backbone. This enables the convolutional kernels to focus on irregular crack damages, thereby enhancing the feature extraction capability for damages. Secondly, to tackle the high algorithm complexity in the feature fusion stage, a lightweight feature fusion module is constructed using GhostConv as a replacement for the C3 module in the original network neck part, reducing the network parameters and complexity. Additionally, to prevent missed detections caused by uneven lighting conditions and shadow obstruction, a lightweight attention mechanism called TripletAttention is introduced in the backbone network to enhance the algorithm's understanding of damage information and context. Finally, experiments conducted on the IEEE open dataset RDD2022 and the Kaggle open dataset Road Damage demonstrate that compared to YOLOv5s, the proposed LDC-YOLOv5 achieves a 1.4% and 4.2% improvement in mAP50 on the two datasets, respectively, with only 67.6% of the model parameters of YOLOv5s.

    参考文献
    相似文献
    引证文献
引用本文

孔令鑫,陈紫强,晋良念,等. 融合可变形卷积的轻量级路面病害检测算法[J]. 科学技术与工程, , ():

复制
分享
文章指标
  • 点击次数:
  • 下载次数:
  • HTML阅读次数:
  • 引用次数:
历史
  • 收稿日期:2024-03-11
  • 最后修改日期:2024-04-19
  • 录用日期:2024-04-25
  • 在线发布日期:
  • 出版日期:
×
亟待确认版面费归属稿件,敬请作者关注