基于Transformer与局部特征融合的轨道紧固件缺陷检测方法

Defect detection method for track fastener based on Transformer and local feature fusion

为解决传统人工巡检轨道交通线路存在的效率低和有安全隐患等问题,提出一种基于Transformer与局部特征融合的轨道紧固件缺陷检测方法。构建轨道紧固件缺陷检测模型,将Transformer与局部特征模块融合,整合局部信息,进而提取轨道紧固件缺陷特征;同时,采用数据增强的方法对轨道紧固件缺陷样本进行数据扩增,扩充数据集,验证所建模型的检测效果。实验结果表明,相较于传统方法,文章提出的方法在识别轨道紧固件缺失和损坏两类缺陷方面的精度和平均准确率均有所提升,在不同的轨道线路实验环境下也表现出良好的检测效果。

To solve the problems of low efficiency and safety hazards in traditional manual inspection of rail transit lines,this paper proposed a rail fastener defect detection method based on Transformer and local feature fusion.The paper constructed a defect detection model for rail fasteners,integrated Transformer with local feature modules,integrated local information,and extracted defect features of rail fasteners,at the same time,used data augmentation methods to expand the dataset of rail fastener defect samples and verify the detection effect of the constructed model.The experimental results show that compared to traditional methods,the proposed method has improved accuracy and average accuracy in identifying two types of defects,namely missed and damaged track fasteners.It also shows good detection performance in different track experimental environments.