基于特征映射贡献度的轻量化路面损伤检测方法

Lightweight pavement damage detection method based on feature mapping contribution degree

  • 摘要: 道路损伤检测在道路维护过程中至关重要,针对目前的道路损伤检测依赖于大型车载检测设备导致其成本过高且普适性差的问题,本文提出了一种应用于边缘设备的超轻量化路面裂纹检测模型。首先构建轻量化的道路损伤检测网络,在此基础上基于特征映射贡献度剪枝策略,删除网络中的冗余参数,保证在边缘设备中进行实时监测的同时不降低检测精度。通过实验对本文提出的模型进行了验证,结果表明本文所提出的模型大小仅有2.06 M,在帧率40的情况下准确率达到了68.6%,相较于原始的YOLOv5s模型,在准确率仅降低3.4%的情况下,参数量降低了70.8%,在边缘平台推理延时降低175 ms。综上所述,本文所提出的模型能够更有效地检测道路特征的同时保持高水平的准确性,对于道路损伤检测具有良好的鲁棒性。本文代码后续公开在https://github.com/835822146/ASE-pruning

     

    Abstract: Road damage detection is very important in the process of road maintenance. In order to solve the problem that the current road damage detection relies on large vehicle detection equipment, which leads to high cost and poor universality, this paper proposes an ultra-lightweight surface crack detection model applied to edge equipment. Firstly, a lightweight road damage detection network is constructed, and on this basis, redundant parameters in the network are deleted based on the feature mapping contribution pruning strategy, so as to ensure real-time monitoring in edge devices without reducing the detection accuracy. Finally, the experimental results show that the size of the proposed model is only 2.06 M, and the accuracy rate reaches 68.6% at the frame rate of 40. Compared with the original YOLOv5s model, when the accuracy is only reduced by 3.4%, the number of parameters is reduced by 70.8%, and the inference delay on the edge platform is reduced by 175 ms. In summary, the model proposed in this paper can detect road characteristics more effectively while maintaining a high level of accuracy, and has good robustness for road damage detection.

     

/

返回文章
返回