高级检索

基于YOLO架构的路面裂缝轻量化检测方法

A lightweight pavement crack detection method based on the YOLO framework

  • 摘要: 针对路面裂缝检测中模型参数过多、计算需求较高的问题,本文提出了一种基于YOLO架构的轻量化改进方法。将原架构的CSPDarknet53主干网络替换为EfficientNet-B0模型,并去除其中的平均池化层和全连接层,以实现网络模型的轻量化。在主干网络和颈部网络中引入BiFormer注意力机制,以减少模型的计算负担。此外,引入CARAFE采样算子,提升模型的特征融合与表征能力。实验结果表明,改进后模型的平均精度(mAP)达到88.7%,帧率达到了86.2 帧/s。该模型在维持较高检测精度的同时,有效降低了网络的权重,参数量与计算量。

     

    Abstract: A lightweight improved method based on the YOLO architecture is proposed to solve the problems of excessive model parameters and high computational complexity in road crack detection.The backbone network CSPDarknet53 of the original architecture is replaced with EfficientNet-B0. The average pooling layer and fully connected layers in EfficientNet-B0 are removed to realize the lightweight design of the entire network model. The BiFormer attention mechanism is introduced into the backbone and neck networks to reduce the computational burden of the model. Furthermore, the CARAFE sampling operator is adopted to improve the feature fusion and representation capabilities of the model. Experimental results show that the mean average precision (mAP) of the improved model reaches 88.7%, and the frame rate reaches 86.2 fps. While maintaining a high detection accuracy, the proposed model reduces the network weight, parameter quantity and computational complexity.

     

/

返回文章
返回