[发明专利]头部姿态估计方法和装置、电子设备和存储介质

说明书

本公开实施例公开了一种头部姿态估计方法和装置、电子设备和存储介质，其中，方法包括：获取驾驶员图像序列，所述驾驶员图像序列包括多帧驾驶员图像；利用第一神经网络对所述驾驶员图像序列中的当前帧图像进行特征提取，得到第一特征；以及利用第二神经网络对所述当前帧图像和前一帧图像之间的光流图像进行特征提取，得到第二特征；将所述第一特征与所述第二特征进行拼接，得到所述当前帧图像对应的拼接特征；利用长短期记忆单元LSTM模型，基于所述当前帧图像对应的拼接特征和前N个历史帧图像对应的拼接特征进行头部姿态预测，得到所述当前帧图像中的头部姿态信息。本公开实施例可以提高头部姿态估计的准确性。

技术领域

本公开涉及人工智能技术，尤其是一种头部姿态估计方法和装置、电子设备和存储介质。

背景技术

随着车辆的不断增多，交通事故也随之增多，当驾驶员在驾驶过程中由于顾及手机等其他事物将其注意力分散到手机等其他事物时，会导致驾驶员无法及时了解道路情况，极易发生道路交通事故，因此，驾驶员的驾驶状态对安全行车的影响非常重要，应尽可能的使驾驶员处于良好的驾驶状态。

随着汽车产业的迅速发展，人们对于汽车智能化的需求越来越大。头部姿态估计是从数字图像或视频图像中推断出人的头部偏转角度的过程，在汽车行驶过程中，通过对汽车驾驶员进行头部姿态估计，可以获知驾驶员是否发生注意力分散，以便避免交通事故，从而提高驾驶的安全性。

在实现本公开的过程中，发明人发现，现有技术通过机器学习算法对驾驶员进行头部姿态估计时，直接以采集的单帧图像提取视觉特征，基于该视觉特征估计出驾驶员的头部姿态，受驾驶室光线条件、汽车行驶稳定性对采集图像的质量影响等因素，无法得到稳定、鲁棒的头部姿态，从而导致头部姿态估计的准确性较低。

发明内容

本公开实施例提供一种头部姿态估计方法和装置、电子设备和存储介质，以提高头部姿态估计的准确性。

本公开实施例的一个方面，提供的一种头部姿态估计方法，包括：

获取驾驶员图像序列，所述驾驶员图像序列包括多帧驾驶员图像；

利用第一神经网络对所述驾驶员图像序列中的当前帧图像进行特征提取，得到第一特征；以及利用第二神经网络对所述当前帧图像和前一帧图像之间的光流图像进行特征提取，得到第二特征；其中，所述前一帧图像为所述驾驶员图像序列中在时序上位于所述当前帧图像之前的一帧图像；

将所述第一特征与所述第二特征进行拼接，得到所述当前帧图像对应的拼接特征；

利用长短期记忆单元LSTM模型，基于所述当前帧图像对应的拼接特征和前N个历史帧图像对应的拼接特征进行头部姿态预测，得到所述当前帧图像中的头部姿态信息；其中，N为大于0的整数。

可选地，在基于本公开上述任一实施例的方法中，所述获取驾驶员图像序列，包括：

通过车辆内至少一个位置部署的红外摄像头进行图像采集，得到所述驾驶员图像序列。

可选地，在基于本公开上述任一实施例的方法中，所述通过红外摄像头进行图像采集，包括：

在所述车辆处于行驶状态时通过红外摄像头进行图像采集；和/或，

在所述车辆的行驶速度超过预设车速时通过红外摄像头进行图像采集；和/或，

在检测到所述车辆点火后通过红外摄像头进行图像采集；和/或，

在检测到所述车辆的启动指令时通过红外摄像头进行图像采集；和/或，

在检测到对所述车辆或所述车辆中部件或系统的控制指令时通过红外摄像头进行图像采集。

可选地，在基于本公开上述任一实施例的方法中，还包括：