[发明专利]基于物体检测的疲劳驾驶检测方法

说明书

本发明公开了一种基于物体检测的疲劳驾驶检测方法，其包括获取驾驶位置处的视频信息，并等间隔选取视频信息中的一帧测试图片；输入测试图片至SSD模型，采用SSD模型计算测试图片与SSD模型内每个类别中每张训练图片之间的损失值；当测试图片与训练图片之间的损失值大于第一设定阈值时，则训练图片对应类别为测试图片的类别；获取SSD模型的全连接层预测的测试图片中驾驶员的位置尺寸，并计算所述测试图片中驾驶员的位置尺寸与上一张测试图片中驾驶员的位置尺寸的差异；当差异小于第二设定阈值，则将计数器数值累加一次，并判断计数器数值是否大于第三设定阈值；若计数器数值大于第三设定阈值时，则提醒驾驶员其已处于疲劳驾驶。

技术领域

本发明涉及图像处理技术领域，具体涉及一种基于物体检测的疲劳驾驶检测方法。

背景技术

当代经济的快速发展提高了人民的生活水平，汽车保有量的不断增长就是一方面的体现，与此同时，交通事故也与日俱增。驾驶员作为道路交通系统中最重要的组成部分，在安全驾驶方面起到决定性的作用。疲劳驾驶作为导致交通事故频发的主要因素之一，如何有效监督驾驶员是否疲劳驾驶以及及时提醒驾驶员注意避免进行疲劳驾驶变得尤为重要。根据我国的《道路交通安全法》规定，连续驾车4个小时以上算是疲劳驾驶。

现阶段对驾驶员进行疲劳驾驶检测的方法大多是对驾驶员的面部提取特征，主要是对驾驶员的眼睛，使用图像处理或信息融合等方法分析驾驶员的状态。这种方法存在以下的问题：一、由于驾驶员疲劳状态下体现出的疲劳特征不尽相同，导致根据单一疲劳驾驶参数无法有效的进行疲劳驾驶检测；二、为了达到精确度，过多的检测算法步骤会降低检测效率。

发明内容

针对现有技术中的上述不足，本发明提供的基于物体检测的疲劳驾驶检测方法采用将驾驶员视频信息与驾驶时间相结合方式进行疲劳驾驶判断。

为了达到上述发明目的，本发明采用的技术方案为：

提供一种基于物体检测的疲劳驾驶检测方法，其包括：

获取驾驶位置处的视频信息，并等间隔选取视频信息中的一帧测试图片；

输入测试图片至SSD模型，采用SSD模型计算测试图片与SSD模型内每个类别中每张训练图片之间的损失值；

当测试图片与训练图片之间的损失值大于第一设定阈值时，则训练图片对应类别为测试图片的类别；

获取SSD模型的全连接层预测的测试图片中驾驶员的位置尺寸，并计算所述测试图片中驾驶员的位置尺寸与上一张测试图片驾驶员的位置尺寸的差异；

当所述差异小于第二设定阈值，则将计数器数值累加一次，并判断计数器数值是否大于第三设定阈值；

若计数器数值大于第三设定阈值时，则提醒驾驶员其已处于疲劳驾驶。

进一步地，当测试图片中驾驶员的位置尺寸大于等于第二设定阈值，且连续两张测试图片中驾驶员的位置尺寸大于等于第二设定阈值时，选取下一张测试图片，同时重置计数器数值。

进一步地，当计数器数值小于等于第三设定阈值时，选取下一张测试图片。

进一步地，输入测试图片至SSD模型，采用SSD模型计算测试图片与SSD模型内每个类别中每张训练图片之间的损失值进一步包括：

将每层卷积层中卷积核所对应特征图划分成若干尺寸相同的特征图单元；

每个特征图单元均以其为中心生成若干默认框，并计算每个默认框的尺度：

其中，s_k为尺度；s_min为尺度的最小值；s_max为尺度的最大值；m为特征图所在卷积层的卷积核数量；k为特征图所对应卷积核在其所在卷积层中的序号；