高级检索

基于动态可变形卷积的轻量化道路缺陷检测方法

Lightweight Road Defect Detection Method Based on Dynamic Deformable Convolution

  • 摘要: 为了提高道路缺陷的检测精度,提出一种基于动态可变形卷积的轻量化道路缺陷实时Transformer检测网络。首先设计了多路径坐标注意力机制模块,并将之与可变形卷积模块深度融合,构建具备可变感受野能力的动态可变形卷积模块,以应对不同形状的道路缺陷;然后在尺度内特征交互模块中引入了可变形注意力机制,以增强对图像中关键目标信息的捕捉与提取效能;最后在跨尺度特征融合模块中引入轻量化模块,以实现整体网络的轻量化。在全球道路缺陷检测挑战赛(GRDDC2020)数据集上与YOLOv5-m等8种网络进行对比实验,结果表明,所提网络的检测效果更加理想,且在网络的参数量与计算量上表现也更加出色;该网络的mAP、参数数量、浮点运算量和检测速度分别为61.1%、19.6M、50.2G和42.3帧/s,可有效地检测道路缺陷,为路面养护工作提供信息。

     

    Abstract: To enhance the precision of road defect detection, a lightweight, real-time Transformer-based detection network incorporating dynamic deformable convolutions has been proposed. Initially, a multi-path coordinate attention mechanism module is developed and seamlessly integrated with deformable convolution modules. This integration creates a dynamic deformable convolution module with variable receptive fields, capable of handling road defects of diverse shapes. Subsequently, a deformable attention mechanism is introduced into the intra-scale feature interaction module, enhancing the network's ability to capture and extract critical information from images. Finally, to achieve an overall reduction in network complexity, lightweight modules are employed in the cross-scale feature fusion stage. Experimental evaluations conducted on the Global Road Defect Detection Challenge (GRDDC2020) dataset compared this network against eight others, including YOLOv5-m. The results indicate that the proposed network not only delivers superior detection performance but also exhibits better efficiency in terms of its parameter count and computational requirements. Specifically, the network achieves a mean Average Precision (mAP) of 61.1%, with 19.6 million parameters and 50.2 GOPS (billion floating-point operations per second). Additionally, it operates at a detection speed of 42.3 frames per second. This makes it effective for detecting road defects and providing essential data for road maintenance efforts.

     

/

返回文章
返回