有效

一种钢轨波磨识别方法、系统及存储介质

钱瑶、徐井芒、魏贤奎、王平、刘泽宇、乐明静、张兴
西南交通大学

摘要

本发明实施例提供一种钢轨波磨识别方法、系统及存储介质,属于铁路轨道检测技术领域。所述钢轨波磨识别方法包括:将预处理后的钢轨波磨真值图像数据送入图像分割识别模型,以提取全局语义特征和局部纹理特征;对所述中间特征提取阶段的输出进行通道平均后经Sobel算子生成边缘特征,并将所述全局语义特征、局部纹理特征和边缘特征融合,获得分割掩码图;将基于所述分割掩码图裁剪的ROI图像片段与所述中间特征提取阶段提取的用于目标检测的特征一并输入目标检测网络,执行基于边界框重叠度优化损失的回归计算,输出钢轨波磨检测结果。本发明方案通过多尺度特征提取与通道空间注意力融合,有效提升钢轨波磨区域的定位精度与分割鲁棒性。