基于Transformer的驾驶习惯预测方法及装置

本发明公开了一种基于Transformer的驾驶习惯预测方法及装置，属于自动驾驶与智能驾驶辅助技术领域，方法为：获取驾驶员历史车辆行驶数据并进行预处理，获得消除不同参数间量纲差异的车辆行驶数据序列；采用滑动窗口在车辆行驶数据序列上进行滑动，提取出包含有对应数据标签的数据样本；采用包含有对应数据标签的数据样本对基于Transformer的驾驶习惯预测模型进行训练，训练过程中，以均方误差损失函数进行反向传播及参数优化，得到训练后的驾驶习惯预测模型；将实时车辆行驶数据输入训练后的驾驶习惯预测模型进行车辆行驶数据实时预测，获得驾驶习惯预测结果。本发明能够因人而异得出辅助驾驶信息，提高驾驶者的驾驶体验。