|
|
一种时空特征融合的端到端自动驾驶方法设计 |
吴武飞 李文波 柏迪 喻军 |
南昌大学信息工程学院 南昌大学数学与计算机学院 |
|
|
摘要 针对现有的端到端自动驾驶模型复杂度高、参数多、计算量大且仅由卷积神经网络组成的网络模型无法处理时间特征等问题,提出了一种融合时空特征的端到端自动驾驶模型M-GRU。该模型由改进的MobileNetV2网络和门控循环单元(GRU)网络组成。其中改进的MobileNetV2网络是在MobileNetV2基础上添加注意力模块,其通过加权处理与驾驶决策密切相关的特征信息,进而提升网络对重要特征的关注。M-GRU模型中的2个网络模块分别提取图像的空间特征和时间特征,通过行为克隆的方式,实现对自动驾驶行为的预测。在模拟器中对M-GRU模型进行训练和测试,并与NVIDIA提出的PilotNet模型进行对比,结果显示M-GRU的损失函数值比PilotNet低,M-GRU模型控制的小车能在道路上完成直行、转弯、加减速任务,同时完成简单和困难模式下的驾驶任务,具有更好的性能。
|
|
关键词 :
自动驾驶,
端到端,
MobileNetV2,
门控循环单元,
注意力机制
|
|
基金资助:国家自然科学基金资助项目(62002147); 中国博士后科学基金资助项目(2020TQ013); |
|
|
|