重庆理工大学学报(自然科学) ›› 2020, Vol. 34 ›› Issue (9): 198-208.doi: 10.3969/j.issn.1674-8425(z).2020.09.024

• 智能技术 • 上一篇    

基于目标特征蒸馏的车道线检测

龙建武,彭 浪,安 勇   

  1. 重庆理工大学计算机科学与工程学院,重庆 400054
  • 收稿日期:2019-10-21 发布日期:2020-10-18
  • 作者简介:龙建武,男,博士,副教授,主要从事图像处理与机器学习研究,Email:jwlong@cqut.edu.cn;通讯作者彭浪,男,硕士研究生,主要从事车道线检测、语义分割研究,Email:5627346@qq.com。
  • 基金资助:
    国家自然科学基金青年基金项目(61502065);重庆市教委人文社科研究(重点)项目(17SKG136);重庆市科委基础科学与前沿技术研究(重点)项目(cstc2015jcyjBX0127);重庆理工大学研究生创新课题项目(ycx20192064;ycx2018247)

  • Received:2019-10-21 Published:2020-10-18

摘要: 为满足自动驾驶系统实时性要求,现有方法将编码器输出的特征图直接上采样进行像素级预测,从而忽略了解码器对于细节特征预测的重要性。为解决该问题,提出一种通用的基于目标特征蒸馏的车道线检测框架。首先,在使用直接上采样方式的网络中,增加一个具有较强特征预测能力的解码器;然后,在网络训练阶段,通过知识蒸馏技术将解码器生成的预测结果作为软目标,以使直接上采样分支学习到更为详尽的车道线信息,让其具有解码器的较强特征预测能力;最后,在网络推理阶段仅需使用直接上采样分支,而无需对解码器进行前向计算,因此相比现有模型在不增加额外计算成本的同时还能提高车道线检测性能。为验证本框架的有效性,将其应用到诸如SCNN、Deeplabv1、ResNet等多种主流的车道线分割方法上。实验结果表明:在不增加额外复杂度的条件下,所提出方法在Culane数据集上获得了更高的F1Measure评分。

关键词: 解码器;蒸馏;车道线检测;分类概率图

中图分类号: 

  • TP391