基于注意力机制与动态蛇形卷积的车道线分割算法

摘要

车道检测作为计算机视觉技术在交通领域的关键应用，具有深远的实用价值。尽管如此，现有的语义分割网络模型在道路语义分割任务中仍面临着精度不足和边缘模糊等挑战。针对这些问题，提出了一种基于UNet模型的改进型车道线图像分割网络架构。在UNet模型的跳跃连接中，引入双重注意力机制，优先突出了车道线的重要性，并有效降低了噪声的干扰。此外，采用了动态蛇形卷积来替代传统的卷积方法，增强了网络对车道线的识别能力。考虑到在曝光不足或光线较暗的背景下进行车道线检测的挑战，在图像预处理阶段引入了一种改进的自适应Gamma校正技术，以增强检测的全面性和准确性。同时，在编码器末端引入了空洞金字塔池化技术。实验结果表明，在满足实时性要求的前提下，该模型TuSimple数据集上达到了98.93%的准确率，相较于其他5种基于语义分割的车道线检测算法，展现出更优越的识别效果，结果验证了应用动态蛇形卷积与双注意力机制改进的有效性。

关键词

车道线检测 / 语义分割 / 注意力机制 / 动态蛇形卷积 / Gamma校正算法

基金资助

黑龙江省“百千万工程”科技重大专项(2021ZX04A01)