基于多分支跨空间注意力模型的列车轴承损伤分类识别方法

摘要文本

本发明公开了一种基于多分支跨空间注意力模型的列车轴承损伤分类识别方法，包括如下步骤：获取列车轮对轴承元件表面图像；根据列车轮对轴承元件表面不同的损伤类型，分别建立图像样本集，并根据损伤类型不同对其进行标签划分；将样本集进行随机划分，确定训练样本集和验证样本集，将训练样本集和验证样本集输入网络模型中训练；使用完成网络模型训练的网络模型对待测的轮对轴承进行损伤识别。构建的网络模型既可以通过多分支模块提取不同语义下更丰富的图像特征信息，又利用跨空间通道连接突出增强重要特征信息，从而能够较为准确地扑捉图像中的关键特征，提升网络的分类识别性能。