【香港科技大学】CLRKDNet：利用知识蒸馏加速车道检测

文章预览

点击下方 卡片 ，关注“ 自动驾驶专栏 ”公众号 自动驾驶干货 ，即可获取 论文链接：https://arxiv.org/pdf/2405.12503 代码和模型链接：https://github.com/weiqingq/CLRKDNet 摘要 本文介绍了CLRKDNet：利用知识蒸馏加速车道检测。道路车道是智能汽车视觉感知系统的组成部分，在安全导航中发挥着关键作用。在车道检测任务中，平衡准确性和实时性是至关重要的，但是现有方法往往会牺牲其中之一。为了解决这种权衡问题，本文引入了CLRKDNet，这是一种平衡检测准确性和实时性的精简模型。最先进的模型CLRNet在各种数据集上均表现出卓越的性能，但是由于其特征金字塔网络（FPN）和多层检测头架构，导致计算开销很大。本文方法简化了FPN结构和检测头，对它们进行重新设计，以将新的教师-学生蒸馏过程与新引入的一系列蒸馏损失相结合。这种结合将推理时间减少 ………………………………