arXiv:2411.10902cs.CV2024-11被引 2

用注意力机制改进U-Net,显著提升车道线检测准确率

Attention-based U-Net Method for Autonomous Lane Detection

  • 在U-Net中加入注意力模块,增强关键特征捕捉能力
  • 新模型达98.98%准确率,远超基础U-Net
  • 适合自动驾驶系统中的高精度车道识别需求

车道检测需精准识别道路车道的位置与形状,是现代辅助及自动驾驶系统的关键技术。然而,车道缺乏明显特征,易被外观相似物体干扰;同时车道数量不固定,线型多样(实线、虚线、单线、双线、合并、分叉等),增加了检测难度。为此,本文提出两种基于深度学习的车道识别方法:第一种采用特征金字塔网络(FPN),实现87.59%的检测准确率;第二种将注意力机制引入U-Net模型,在语义分割任务中表现更优。该改进模型达到98.98%准确率,显著超越基础U-Net,在对比分析中展现出明显优势。研究成果为开发更高效可靠的车道检测方法提供了新路径,显著提升了自动驾驶系统的能力。

原文摘要 · Abstract (English)

Lane detection involves identifying lanes on the road and accurately determining their location and shape. This is a crucial technique for modern assisted and autonomous driving systems. However, several unique properties of lanes pose challenges for detection methods. The lack of distinctive features can cause lane detection algorithms to be confused by other objects with similar appearances. Additionally, the varying number of lanes and the diversity in lane line patterns, such as solid, broken, single, double, merging, and splitting lines, further complicate the task. To address these challenges, Deep Learning (DL) approaches can be employed in various ways. Merging DL models with an attention mechanism has recently surfaced as a new approach. In this context, two deep learning-based lane recognition methods are proposed in this study. The first method employs the Feature Pyramid Network (FPN) model, delivering an impressive 87.59% accuracy in detecting road lanes. The second method, which incorporates attention layers into the U-Net model, significantly boosts the performance of semantic segmentation tasks. The advanced model, achieving an extraordinary 98.98% accuracy and far surpassing the basic U-Net model, clearly showcases its superiority over existing methods in a comparative analysis. The groundbreaking findings of this research pave the way for the development of more effective and reliable road lane detection methods, significantly advancing the capabilities of modern assisted and autonomous driving systems.

车道检测注意力机制U-Net自动驾驶

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。