武汉理工大学学报(交通科学与工程版)2024,Vol.48Issue(2) :380-384,391.DOI:10.3963/j.issn.2095-3844.2024.02.031

基于深度学习的预制梁表面气泡缺陷检测

Deep Learning-based Detection of Bubble Defects on the Surface of Precast Beams

陈烨 夏文俊 钱家辉 赵章焰
武汉理工大学学报(交通科学与工程版)2024,Vol.48Issue(2) :380-384,391.DOI:10.3963/j.issn.2095-3844.2024.02.031

基于深度学习的预制梁表面气泡缺陷检测

Deep Learning-based Detection of Bubble Defects on the Surface of Precast Beams

陈烨 1夏文俊 2钱家辉 1赵章焰1
扫码查看

作者信息

  • 1. 武汉理工大学交通与物流工程学院 武汉 430063
  • 2. 中铁第四勘察设计院集团有限公司 武汉 430063
  • 折叠

摘要

文中提出一种基于YOLOv5s的预制梁表面气泡缺陷检测算法.该算法在原模型的基础上引入CBAM注意力模块,增强通道间信息的关联性及兴趣特征的关注度;在颈部网络中用BiFPN加权双向金字塔结构,改进网络特征融合模块,实现快速的多尺度特征融合.在检出气泡缺陷后,提出基于面积和直径的两个评价指标对气泡进行分类.结果表明:改进模型具有更强的特征提取能力,平均检测精度(mAP)为95.8%,相对于原模型提高了 2.3%,准确率提高了 6.5%,召回率提高了 3.5%,在气泡缺陷检测任务中有效减少了漏检和误检,具备更好的检测性能.

Abstract

An algorithm for detecting bubble defects on the surface of precast beams based on YOLOv5s was proposed.Based on the original model,the algorithm introduced the CBAM attention module to enhance the relevance of information between channels and the attention of interest features.In the neck network,BiFPN weighted bidirectional pyramid structure was used to improve the network fea-ture fusion module and realize fast multi-scale feature fusion.After detecting bubble defects,two e-valuation indexes based on area and diameter were proposed to classify bubbles.The results show that the improved model has stronger feature extraction ability,the average detection accuracy(mAP)is 95.8%,which is 2.3%higher than the original model,the accuracy is 6.5%higher,and the recall is 3.5%higher.In the task of bubble defect detection,the missed detection and false detection are effec-tively reduced,and the detection performance is better.

关键词

预制梁/气泡缺陷/YOLOv5s/注意力机制/BiFPN

Key words

precast beams/bubble defects/YOLOv5s/attention mechanism/BiFPN

引用本文复制引用

基金项目

中交集团首个揭榜挂帅科技攻关项目(2021-ZJKJ-JBGS01)

出版年

2024
武汉理工大学学报(交通科学与工程版)
武汉理工大学

武汉理工大学学报(交通科学与工程版)

CSTPCD
影响因子:0.462
ISSN:2095-3844
参考文献量8
段落导航相关论文