钢轨缺陷检测方法、装置、设备及存储介质

本发明提供了一种钢轨缺陷检测方法、装置、设备及存储介质，属于图像识别技术领域，其方法包括：对钢轨深度图进行分割，获得多个局部切片；基于特征提取模块对各局部切片进行特征提取，获得多个初始特征；基于特征重嵌入模块对多个初始特征进行位置重嵌入，获得多个重嵌入特征；基于多实例学习模块对多个重嵌入特征进行缺陷检测，获得钢轨级别的检测结果。本发明通过钢轨深度图进行缺陷检测，显著降低钢轨缺陷检测方法的误报率，提高检测准确性。并且，采用切片化处理显著减少单次输入的数据量，降低计算复杂度。同时，采用多实例学习框架，仅需弱监督的钢轨级标签即可训练模型，提升小样本场景下的检测性能。