一种基于自融合机制的钢轨表面缺陷定位方法及相关装置

本发明提供了一种基于自融合机制的钢轨表面缺陷定位方法及相关装置，涉及智能检测技术领域，包括获取钢轨表面伤损的图像数据和正常钢轨的图像数据；在缩放后的二维图像中标记出伤损位置，得到真值图像；将高程图像与对应的二维图像合并为样本图像；构建数据集，将所述数据集划分为训练集和测试集；构建自融合机制的卷积神经网络模型，利用所述训练集和测试集对卷积神经网络模型进行训练和测试，得到训练完成的钢轨表面缺陷定位模型；将待检测钢轨的二维图像和高程图像合并后输入至钢轨表面缺陷定位模型，得到钢轨表面的伤损位置，本发明用于解决现有的语义分割算法用于钢轨表面缺陷的检测时定位不准确的技术问题。