一种基于轻量级网络的轨道提取方法

本发明属于轨道交通信息技术领域，提供了一种基于轻量级网络的轨道提取方法，包括以下步骤：步骤1、采用轻量级网络训练轨道数据集，获得轨道识别模型；步骤2、采集轨道实时图像，输入至轨道识别模型中，获取轨道坐标；本发明通过深度可分离卷积显著减少了模型的参数量和计算复杂度，使其在实时处理轨道图像时具备更快的推理速度。此外，该模型在多种环境下的表现良好，能够提升轨道识别的准确性和鲁棒性。