摘要
本发明公开了一种基于Transformer编码器和空洞卷积的车道线检测方法,包括:采用无监督风格迁移将白天道路交通图像转换成夜间交通图像;构建主干特征提取网络,将普通卷积替换成空洞卷积以提取车道线的局部特征;建立Transformer编码器来提取车道线的全局特征;对提取到的局部特征和全局特征利用双向特征金字塔进行加权融合,并利用网络学习权重的大小来优化特征融合;构建车道线检测头;对模型进行训练,使模型收敛获得车道线检测网络参数;将模型安装在车载摄像头上,用于对车道线进行实时检测,得到车道线实例分割图。该方法可以提高不同场景下车道线特征提取的精度与计算效率,同时可以方便地整合到其他现有车道线检测算法中进行端到端训练。
- 单位