4 个月前

CLRKDNet:利用知识蒸馏加速车道检测

CLRKDNet:利用知识蒸馏加速车道检测

摘要

车道是智能车辆视觉感知系统的重要组成部分,在安全导航中发挥着关键作用。在车道检测任务中,平衡精度与实时性能至关重要,但现有的方法往往在这两者之间做出妥协。为了解决这一权衡问题,我们引入了CLRKDNet,这是一种简化模型,能够在保证检测精度的同时实现实时性能。最先进的模型CLRNet已经在多个数据集上展示了卓越的性能,但由于其特征金字塔网络(Feature Pyramid Network, FPN)和多层检测头架构的存在,计算开销较大。我们的方法简化了FPN结构和检测头,并重新设计了它们,结合了一种新颖的教师-学生蒸馏过程以及新引入的一系列蒸馏损失。这种组合将推理时间最多减少了60%,同时保持了与CLRNet相当的检测精度。这种对精度和速度的战略平衡使得CLRKDNet成为自动驾驶应用中实时车道检测任务的一个可行解决方案。

代码仓库

weiqingq/CLRKDNet
官方
pytorch

基准测试

基准方法指标
lane-detection-on-culaneCLRKDNet (ResNet-18)
F1 score: 79.66
lane-detection-on-culaneCLRKDNet (DLA-34)
F1 score: 80.68

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
CLRKDNet:利用知识蒸馏加速车道检测 | 论文 | HyperAI超神经