[发明专利]用于识别驾驶行为的方法和装置

说明书

本申请实施例公开了用于识别驾驶行为的方法和装置。该方法的一具体实施方式包括获取目标图像；从目标图像中分割出目标人体框，其中，目标图像中包括目标人体框；基于目标人体框和预先训练的驾驶行为识别模型，得到目标人体框对应的目标驾驶行为识别结果，其中，驾驶行为识别模型用于识别人体框的驾驶行为类别；基于目标驾驶行为识别结果确定驾驶行为。该实施方式提高了对驾驶行为的识别准确度。

技术领域

本申请实施例涉及图像识别技术领域，具体涉及用于识别驾驶行为的方法和装置。

背景技术

对于公共运营车辆，如出租车、公交车、长途大巴车等，由于涉及众多乘客安全，驾驶员驾驶安全性显得尤为重要。因此，许多公共运行车辆都已安装车载监控摄像头，以方便对应的公司或监管部门对驾驶员驾驶行为进行监控。对于驾驶员经常出现的一些危险驾驶行为，如抽烟、打电话、疲劳驾驶等，需要及时发现并进行警告，最大限度保证车辆行驶安全。

目前，多采用人眼对车载监控摄像头拍摄的监控录像抽样观看的方式识别是否存在危险驾驶行为。

发明内容

本申请实施例提出了用于识别驾驶行为的方法和装置。

第一方面，本申请实施例提供了一种用于识别驾驶行为的方法，包括：获取目标图像；从目标图像中分割出目标人体框，其中，目标图像中包括目标人体框；基于目标人体框和预先训练的驾驶行为识别模型，得到目标人体框对应的目标驾驶行为识别结果，其中，驾驶行为识别模型用于识别人体框的驾驶行为类别；基于目标驾驶行为识别结果确定驾驶行为。

在一些实施例中，从目标图像中分割出目标人体框，包括：基于目标图像和预先训练的人体检测模型，得到目标图像对应的目标人体检测结果，其中，人体检测模型用于检测图像中的人体框的位置信息和人体类别；基于目标人体检测结果分割目标图像，得到目标人体框。

在一些实施例中，基于目标图像和预先训练的人体检测模型，得到目标图像对应的目标人体检测结果，包括：对目标图像进行采样，得到采样图像；修改采样图像的像素点的像素值，得到修改图像。将修改图像输入至人体检测模型，得到目标人体检测结果。

在一些实施例中，基于目标人体检测结果分割目标图像，得到目标人体框，包括：若待检测图像中存在至少两个类别相同的人体框，基于面积从至少两个类别相同的人体框中确定目标人体框；从目标图像中分割出目标人体框。

在一些实施例中，基于目标人体框和预先训练的驾驶行为识别模型，得到目标人体框对应的目标驾驶行为识别结果，包括：对目标人体框进行采样，得到采样人体框；将采样人体框输入至驾驶行为识别模型，得到目标驾驶行为识别结果。

在一些实施例中，目标驾驶行为识别结果包括目标人体框对应的预设的驾驶行为类别集合中的驾驶行为类别的置信度；以及基于目标驾驶行为识别结果确定驾驶行为，包括：对于驾驶行为类别集合中的驾驶行为类别，计算目标人体框对应该驾驶行为类别的平均置信度；将平均置信度与预设置信度阈值进行比较，基于比较结果确定驾驶行为。

在一些实施例中，人体检测模型通过如下步骤训练得到：获取第一训练样本集合，其中，第一训练样本包括样本图像和样本图像对应的样本人体标注结果；将第一训练样本集合中的第一训练样本中的样本图像作为输入，将输入的样本图像对应的样本人体标注结果作为输出，训练得到人体检测模型。

在一些实施例中，驾驶行为识别模型通过如下步骤训练得到：获取第二训练样本集合，其中，第二训练样本包括样本人体框和样本人体框对应的样本驾驶行为标注结果；将第二训练样本集合中的第二训练样本中的样本人体框作为输入，将输入的样本人体框对应的样本驾驶行为标注结果作为输出，训练得到驾驶行为识别模型。