一种时空特征融合的端到端自动驾驶方法设计

吴武飞; 李文波; 柏迪; 喻军

doi:10.13764/j.cnki.ncdg.2024.02.010

南昌大学学报（工科版） ›› 2024, Vol. 46 ›› Issue (02) : 162 -169. DOI: 10.13764/j.cnki.ncdg.2024.02.010

一种时空特征融合的端到端自动驾驶方法设计

吴武飞, 李文波, 柏迪, 喻军

作者信息 +

Author information +

文章历史 +

PDF

摘要

针对现有的端到端自动驾驶模型复杂度高、参数多、计算量大且仅由卷积神经网络组成的网络模型无法处理时间特征等问题，提出了一种融合时空特征的端到端自动驾驶模型M-GRU。该模型由改进的MobileNetV2网络和门控循环单元(GRU)网络组成。其中改进的MobileNetV2网络是在MobileNetV2基础上添加注意力模块，其通过加权处理与驾驶决策密切相关的特征信息，进而提升网络对重要特征的关注。M-GRU模型中的2个网络模块分别提取图像的空间特征和时间特征，通过行为克隆的方式，实现对自动驾驶行为的预测。在模拟器中对M-GRU模型进行训练和测试，并与NVIDIA提出的PilotNet模型进行对比，结果显示M-GRU的损失函数值比PilotNet低，M-GRU模型控制的小车能在道路上完成直行、转弯、加减速任务，同时完成简单和困难模式下的驾驶任务，具有更好的性能。