首页 | 本学科首页   官方微博 | 高级检索  
     

基于自注意力机制的桥梁螺栓检测算法
引用本文:鞠晓臣,赵欣欣,钱胜胜.基于自注意力机制的桥梁螺栓检测算法[J].浙江大学学报(自然科学版 ),2022,56(5):901-908.
作者姓名:鞠晓臣  赵欣欣  钱胜胜
作者单位:1. 中国铁道科学研究院集团有限公司 铁道建筑研究所,北京 1000812. 中国科学院 自动化研究所,北京 100190
基金项目:高铁联合基金资助项目(U1934209);中国铁路总公司系统性重大课题资助项目(P2018G002);中国铁道科学研究院集团有限公司科研项目重大课题(2020YJ087);安徽省引江济淮集团公司科研项目(YJJH-ZT-ZX-29210923429)
摘    要:基于构建的真实桥梁螺栓场景数据集,提出基于自注意力机制与中心点回归(SACPR)的螺栓检测算法. 构建基于真实场景的高质量桥梁螺栓场景数据集,并针对数据不均衡、多样性不够的问题,使用数据增强方法进行数据扩充,从而获得更高的分类精度. 采用基于深度学习框架的SACPR算法检测不同场景下的螺栓,并进行标示. 对螺栓检测准确率进行验证实验,验证SACP算法的有效性. 将试验结果与YOLOv3、Faster-RCNN、RetinaNet这3种算法结果进行对比,发现3种检测方法的识别精度分别为80.56 %、87.71%、93.89%,而所提出的SACPR 算法的识别精度为93.91%,明显优于YOLOv3算法和Faster-RCNN算法;虽然SACPR算法与RetinaNet算法的识别精度较接近,但前者的检测速度是后者的5.6倍.

关 键 词:桥梁  图像识别  SACPR  螺栓检测  自注意力机制  

Self-attention mechanism based bridge bolt detection algorithm
Xiao-chen JU,Xin-xin ZHAO,Sheng-sheng QIAN.Self-attention mechanism based bridge bolt detection algorithm[J].Journal of Zhejiang University(Engineering Science),2022,56(5):901-908.
Authors:Xiao-chen JU  Xin-xin ZHAO  Sheng-sheng QIAN
Abstract:A bolt detection model algorithm based on self attention mechanism and center point regression (SACPR) was proposed based on the real bridge bolt scene data set. Firstly, a high-quality bridge bolt scene data set based on the real scene was constructed, and for the problems of data imbalance and insufficient diversity, data enhancement method was used to expand the data, so as to obtain higher classification accuracy. Secondly, SACPR model algorithm based on deep learning framework was used to detect bolts in different scenes, and label them. Finally, the validity of the proposed method was verified by the verification experiment of bolt detection accuracy. Comparison was conducted with the results of YOLOv3, Faster-RCNN and RetinaNet, and results showed that the recognition accuracy of the three detection methods was 80.56%, 87.71% and 93.89% respectively, while the recognition accuracy of SACPR model method was 93.91%. The accuracy of SACPR model method was obviously better than that of YOLOv3 model algorithm and Faster-RCNN model algorithm. Although the recognition accuracy was almost the same as that of RetinaNet model algorithm, the detection speed of SACPR model method was 5.6 times of that of RetinaNet model.
Keywords:bridge  image recognition  SACPR  bolt detection  self-attention mechanism  
点击此处可从《浙江大学学报(自然科学版 )》浏览原始摘要信息
点击此处可从《浙江大学学报(自然科学版 )》下载全文
设为首页 | 免责声明 | 关于勤云 | 加入收藏

Copyright©北京勤云科技发展有限公司  京ICP备09084417号