基于注意力增强双路径网络的铁路客车轴箱盖螺栓故障检测方法

一种基于注意力增强双路径网络的铁路客车轴箱盖螺栓故障检测方法，属于铁路客车故障检测技术领域。本发明针对轴箱盖螺栓在密集复杂背景环境下不易准确识别的问题。包括：构建双路径神经网络模型采用全局特征路径获取轴箱盖宏观几何特征；采用局部细节特征路径得到增强后图像微观特征；采用特征融合模块进行注意力融合，得到融合后特征图像，并得到轴箱盖类型、轴箱盖方向及螺栓状态初步识别结果；再结合预定义掩膜矩阵，采用空间姿态匹配算法将预定义掩膜矩阵变换至待识别图像轴箱盖空间进行空间匹配对齐，通过预定义掩膜矩阵与训练后双路径神经网络模型输出的螺栓状态初步识别结果标签的交集运算，确定螺栓最终状态。本发明用于螺栓故障检测。