[发明专利]一种用于辅助驾驶的视线检测方法、装置及存储介质

说明书

本发明提供了一种用于辅助驾驶的视线检测方法、装置及存储介质，其可以获取可靠的视线数据，计算操作方便，提升了视线检测的准确率，计算复杂度较低,其通过相机采集数据，使得被采集人员的视线跟随目标物体移动，获得人眼跟随目标物体移动的图像，计算视线方向数据；构建包括眼球模型推理模块和视线回归模块的视线检测模型，将图像输入眼球模型推理模块输出眼球二值图，将眼球二值图输入视线回归模块，输出推理的视线方向；由采集的人眼随目标物体移动的图像和视线方向数据构建视线数据集，通过视线数据集训练视线检测模型，获得训练好的视线检测模型；将采集的待检测的人眼图像输入训练好的视线检测模型，输出待检测的人眼的视线方向。

技术领域

本发明涉及辅助驾驶和人工智能技术领域，具体涉及一种用于辅助驾驶的视线检测方法、装置及存储介质。

背景技术

现如今,随着人们生活水平的不断提高,汽车已经逐渐成为人们生活中的必备品,而安全驾驶也越来越被人们所关注,因此,安全驾驶的研究也越来越深入。在行车驾驶中影响因素最大之一就是驾驶员的视线方向，通过监控驾驶员的视线方向，可以提高驾驶员的注意力，提高交通驾驶的安全性。

目前的视线检测技术，有以下两个难点：样本采集困难和视线检测结果不准确。目前的公开数据集，基于红外相机的视线数据集几乎没有，而且大部分是欧美人眼，亚洲人眼的视线数据集几乎没有。视线数据和检测与分类的数据集不同，难以人工标注，需要重新进行视线数据的采集，而空间中视线数据是三维信息，不好采集，视线检测结果的准确性受数据集、模型的影响，缺乏可靠数据集的情况下，会导致视线检测结果也不理想，同时一些现有的方法也存在计算复杂度较高的问题。

发明内容

针对上述问题，本发明提供了一种用于辅助驾驶的视线检测方法、装置及存储介质，其可以获取可靠的视线数据作为样本，计算操作方便，提升了视线检测的准确率，计算复杂度较低。

其技术方案是这样的：一种用于辅助驾驶的视线检测方法，其特征在于，包括以下步骤：通过相机采集数据，使得被采集人员的视线跟随目标物体移动，获得人眼跟随目标物体移动的图像；

根据获得的人眼跟随目标物体移动的图像，计算视线方向数据；

构建视线检测模型，所述视线检测模型包括眼球模型推理模块和视线回归模块，将图像输入眼球模型推理模块，输出眼球二值图，所述眼球二值图包括眼球轮廓和瞳孔轮廓，将眼球二值图输入视线回归模块，输出推理的视线方向；

由采集的人眼随目标物体移动的图像和计算得到的视线方向数据构建视线数据集，通过视线数据集训练所述视线检测模型，直至模型收敛，获得训练好的视线检测模型；

将采集的待检测的人眼的图像输入训练好的视线检测模型，输出待检测的人眼的视线方向。

进一步的，所述的通过相机采集数据，包括：将红外相机和深度相机放置在同一水平面内，在被采集人员面前摆放目标物体，移动目标物体，被采集人员通过人眼观察目标物体，视线跟随目标物体移动，通过红外相机和深度相机同步采集图像，使得目标物体移动的轨迹铺满图像的整个区域。

进一步的，所述的被采集人员通过人眼观察目标物体，包括：被采集人员的头部姿势，维持面部朝向一个方向不变，移动目标物体，被采集人员通过人眼观察目标物体，视线跟随目标物体移动，通过红外相机和深度相机采集图像，使得目标物体移动的轨迹铺满图像的整个区域。

进一步的，所述的被采集人员通过人眼观察目标物体，包括：被采集人员的视线跟随目标物体移动，被采集人员的头部与视线的方向一致，随目标物体一起移动，通过红外相机和深度相机采集图像，使得目标物体移动的轨迹铺满图像的整个区域。

进一步的，通过红外相机和深度相机采集图像中，至少包括左上方、上方、右上方、左方、前方、右方、左下方、下方和右下方的面部朝向的图像。

进一步的，所述目标物体采用球形物体。