首页|基于B-YOLOv5的轻量化裂缝检测算法

基于B-YOLOv5的轻量化裂缝检测算法

扫码查看
针对当前公路路面缺陷检测算法存在的特征提取不完善且难以部署到嵌入式设备上、细小裂纹及凹坑漏检等问题,以YOLOv5算法为基础,使用DepthSepConv模块代替原有的C3结构,把原有的CSPDarknet53主干网络改进成了更加轻量化的网络结构,结合BIFPN特征融合思想,将原来的PANet路径融合结构改进为一种更有效的带权重的B-PANet特征融合结构.试验结果表明,所改进的B-YOLOv5算法在相同的数据集和试验条件下,不仅精度提高了5.81%、检测速度提升两倍,还可降低细小裂纹和凹坑的漏检率,模型参数大小也仅仅是YOLOv5的八分之一.B-YOLOv5算法完全可以满足实时性的需要,且可更好地部署在Jetson Xavier NX嵌入式设备上.
Lightweight crack detection algorithm based on B-YOLOv5
In view of the current highway pavement defect detection algorithm feature extraction is imperfect and difficult to deploy on embedded equipment,missing detection of tiny cracks and pits,this paper used the Depth Sep Conv module instead of the original C3 structure,the original CSP Darknet 53 backbone network was improved into more lightweight network structure by combining with BIFPN feature fusion ideas,the original PANet path fusion structure was improved to be a more effective weight B-PANet feature fusion structure.The experimental results showed that the B-YOLOv5 algorithm improved in this paper can not only improve the accuracy of 5.81% and double the detection speed under the same data set and experimental conditions,but also improve the missed detection problem of fine cracks and pits,and the parameter size of the model was only one eighth of YOLOv5.The B-YOLOv5 algorithm can fully meet the needs of real-time performance and be better deployed on embedded devices.

crack detectionBIFPNYOLOv5feature fusionB-YOLOv5

胡肖肖、雷斌、蒋林、段宛妮

展开 >

武汉科技大学 机械自动化学院,武汉 430081

武汉科技大学 机器人与智能系统研究院,武汉 430081

裂缝检测 BIFPN YOLOv5 特征融合 B-YOLOv5

国家重点研发计划

2019YFB1310000

2024

无损检测
中国机械工程学会 上海材料研究所

无损检测

CSTPCD
影响因子:0.558
ISSN:1000-6656
年,卷(期):2024.46(7)
  • 6