一种钢轨波磨的检测方法、装置、设备及可读存储介质

本发明提供了一种钢轨波磨的检测方法、装置、设备及可读存储介质，涉及图像处理技术领域，包括获取历史时间段内的铁路钢轨图像；对铁路钢轨图像进行图像预处理；基于目标钢轨图像和钢轨特征图像训练初始深度学习模型，得到图像分割模型；得到图像识别模型；将目标钢轨的钢轨图像输入图像分割模型中；将目标特征图像输入图像识别模型中，得到目标钢轨的钢轨波磨检测结果。本发明通过在级联空间金字塔池化结构中融入适用于钢轨波磨特征提取的编码模块，在解码阶段多重特征融合时引入高效的注意力机制模块，从而进行有效的空间交互，增强了通道维度上的期望表征，使得模型能够实现对钢轨波磨状态的自动准确检测。