用扩散模型检测车道线,通过逐步去噪生成精准车道坐标。
DiffusionLane: Diffusion Model for Lane Detection
- 将车道检测建模为参数空间的去噪过程,逐步优化车道起点与角度。
- 在CULane上达81.32% F1,在Tusimple上96.89%准确率,超越现有方法。
- 混合解码器+辅助头提升特征表达,适合自动驾驶车道识别场景。
本文提出一种基于扩散模型的车道检测方法DiffusionLane,将车道检测视为参数空间中的去噪过程。首先,对真实车道的参数(起点和角度)添加高斯噪声生成带噪车道锚点,模型通过渐进式去噪还原目标车道。其次,针对编码器因噪声锚点导致的特征表示不佳问题,设计混合解码策略,结合全局与局部解码器生成高质量车道。同时,在训练阶段引入辅助头,利用可学习车道锚点增强编码器监督。在Carlane、Tusimple、CULane和LLAMAS四个基准上的实验表明,DiffusionLane具备强泛化能力与优异性能:使用ResNet18时在跨域数据集Carlane上至少提升1%准确率;使用MobileNetV4在CULane上取得81.32% F1,在Tusimple上以ResNet34达96.89%准确率,在LLAMAS上以ResNet101达97.59% F1。代码将开源。
原文摘要 · Abstract (English)
In this paper, we present a novel diffusion-based model for lane detection, called DiffusionLane, which treats the lane detection task as a denoising diffusion process in the parameter space of the lane. Firstly, we add the Gaussian noise to the parameters (the starting point and the angle) of ground truth lanes to obtain noisy lane anchors, and the model learns to refine the noisy lane anchors in a progressive way to obtain the target lanes. Secondly, we propose a hybrid decoding strategy to address the poor feature representation of the encoder, resulting from the noisy lane anchors. Specifically, we design a hybrid diffusion decoder to combine global-level and local-level decoders for high-quality lane anchors. Then, to improve the feature representation of the encoder, we employ an auxiliary head in the training stage to adopt the learnable lane anchors for enriching the supervision on the encoder. Experimental results on four benchmarks, Carlane, Tusimple, CULane, and LLAMAS, show that DiffusionLane possesses a strong generalization ability and promising detection performance compared to the previous state-of-the-art methods. For example, DiffusionLane with ResNet18 surpasses the existing methods by at least 1\% accuracy on the domain adaptation dataset Carlane. Besides, DiffusionLane with MobileNetV4 gets 81.32\% F1 score on CULane, 96.89\% accuracy on Tusimple with ResNet34, and 97.59\% F1 score on LLAMAS with ResNet101. Code will be available at https://github.com/zkyntu/UnLanedet.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。