一种轨道线识别方法、装置、电子设备和可读存储介质

本申请属于计算机视觉技术领域，主要提供了一种轨道线识别方法、装置、电子设备和可读存储介质，本申请通过图像编码器对输入的待检测轨道线图像进行特征提取，得到图像特征，接着，通过图像解码器基于图像特征采用可变形注意力机制和多头自注意力机制对层次查询特征向量进行多次迭代更新，得到蕴含长距离上下文信息的解码特征，再接着通过预测模块基于所述解码特征输出预设条数的轨道线的预测置信度以及每条轨道线的点集坐标，最后，得到待检测轨道线图像中包含的一条或多条轨道线，实现了端到端的轨道线识别，对于轨道线结构较复杂的场景，能够准确地进行轨道线识别，解决了现有的方案存在的复杂环境适应性弱、精度不足的问题。