一种钢轨表面与接触网表面图像分割方法及装置

摘要

本发明实施例公开一种钢轨表面与接触网表面图像分割方法及装置，方法包括：获取目标图像；将目标图像输入深度神经网络分割模型，输出钢轨区域的分割图和接触网区域的分割图；其中，深度神经网络分割模型是基于获取样本图像对深度神经网络进行训练后得到的，深度神经网络中使用的卷积均使用5×3卷积。本发明能实现高精度、强鲁棒的钢轨表面和接触网表面区域的图像提取，速度快，适应能力强。