[发明专利]一种动作识别模型构建方法、装置、存储介质及电子设备

说明书

本发明公开了一种动作识别模型构建方法、装置、存储介质及电子设备，所述方法包括：获取图像训练样本并将所述图像训练样本输入至目标神经网络；利用所述目标神经网络中的检测分支网络对所述图像训练样本中每一张图像进行处理得到对应的第一方向向量、第二方向向量、第三方向向量；根据每一张图像的目标人物的动作状态和所述第一方向向量、第二方向向量、第三方向向量对预设神经网络模型进行训练得到动作识别模型。该方法通过目标神经网络可以达到实时的性能；通过每一张图像的动作状态以及对应的方向向量进行训练得到的动作识别模型不需要进行复杂的后处理，可以直接识别细粒度的动作。

技术领域

本发明涉及汽车驾驶辅助终端技术领域，具体涉及一种动作识别模型构建方法、装置、存储介质及电子设备。

背景技术

手离方向盘检测(hands-off detection)和安全带佩戴检测(safe beltfastness detection)是座舱司机状态检测重要的两个方面。基于传统的传感器判断是否手扶方向盘或佩戴安全带，容易受人为干扰，比如，基于压力感应的方向盘，可以采用力道合适的夹子或外人干扰，同理安全带佩戴也易受人为干扰，目前基于视觉的识别方案已成为主流。现有基于视觉的识别方案，都是通过识别物体(手、安全带、方向盘)，再通过复杂的后处理逻辑进行判断是否手离方向盘或是否佩戴安全带，但是后处理逻辑复杂且难以区分细粒度的动作状态。

发明内容

有鉴于此，本发明实施例提供了涉及一种动作识别模型构建方法、装置、存储介质及电子设备，以解决现有技术中通过动作识别模型输出识别结果时需要进行复杂的后处理逻辑且难以区分细粒度的动作状态的技术问题。

本发明提出的技术方案如下：

本发明实施例第一方面提供一种动作识别模型构建方法，该动作识别模型构建方法包括：获取图像训练样本并将所述图像训练样本输入至目标神经网络；利用所述目标神经网络中的检测分支网络对所述图像训练样本中每一张图像中包含的目标人物整个身体、待识别动作对应的目标人物关键部位以及待识别动作作用物体分别进行检测得到所述目标人物整个身体的多个身体中心点、所述待识别动作对应的目标人物关键部位的多个关键部位中心点以及所述待识别动作作用物体的多个物体中心点，每一个身体中心点、关键部位中心点以及物体中心点都对应一个方向向量；根据所述多个身体中心点及每一个身体中心点对应的方向向量、所述多个关键部位中心点及每一个关键部位中心点对应的方向向量、所述多个物体中心点及其每一个物体中心点对应的方向向量，利用所述目标神经网络中的匹配网络进行回归匹配处理，得到唯一身体中心点及其对应的方向向量、唯一关键部位中心点及其对应的方向向量以及唯一物体中心点及其对应的方向向量；根据所述唯一关键部位中心点及其对应的方向向量与所述唯一物体中心点及其对应的方向向量确定所述待识别动作对应的目标人物身体作用点距离作用物体的距离中心点及其对应的方向向量；根据所述距离中心点及其对应的方向向量和所述唯一关键部位中心点及其对应的方向向量确定所述距离中心点与所述唯一关键部位中心点的第一方向向量，根据所述距离中心点及其对应的方向向量和所述唯一物体中心点及其对应的方向向量确定所述距离中心点与所述唯一物体中心点的第二方向向量，根据所述唯一身体中心点及其对应的方向向量和所述唯一关键部位中心点及其对应的方向向量确定所述唯一身体中心点与所述唯一关键部位中心点之间的第三方向向量；根据每一张图像的目标人物的动作状态和所述第一方向向量、第二方向向量、第三方向向量对预设神经网络模型进行训练得到动作识别模型。