[发明专利]一种基于图像增强技术的疲劳驾驶检测方法

说明书

本发明公开了一种基于图像增强技术的疲劳驾驶检测方法，包括以下步骤，S1：采用摄像头对驾驶图像进行采集；S2：驾驶图像中定位驾驶员的眼睛区域，判断驾驶员是否佩戴墨镜，并获得粗定位图像；S3：根据不同光线情况和驾驶员是否佩戴墨镜对粗定位图像进行处理得到面部轮廓图像；S4：将面部轮廓图像归一化，传入预先训练好的EnlightenGAN模型进行光照增强，并转化得到灰度图像；S5：从灰度图像中提取人脸中68个关键点的坐标，并将上述关键点坐标转换成numpy数组；S6：从numpy数组中提取左右眼的点坐标并计算眼睛纵横比，当任一只眼睛的纵横比在一段时间中低于预先设定的阈值时，则发出警报来提醒司机；优点是检测准确率高且降低了数据采集的难度。

技术领域

本发明涉及疲劳驾驶检测技术领域，尤其涉及一种基于图像增强技术的疲劳驾驶检测方法。

背景技术

随着国家汽车行业的发展及国民经济收入的快速增长,人均拥有汽车的数量急剧增加,国民的出行安全问题也日益突出。据不完全统计,约有60％的重大交通事故是由驾驶员疲劳驾驶引发的。因此,对于驾驶员疲劳状态和道路信息的检测,成为防范车辆安全行驶问题的关键。

基于计算机视觉技术对驾驶员进行疲劳驾驶检测是目前的主流方案，但是对于夜晚行车等情况，一般的摄像设备并不能很好地满足检测需求，存在误判的可能性。

发明内容

为了解决上述现有技术中存在的不足，本发明提供一种基于图像增强技术的疲劳驾驶检测方法，其检测准确率高且降低了数据采集的难度。

本发明解决上述技术问题所采用的技术方案为：一种基于图像增强技术的疲劳驾驶检测方法，包括以下步骤，

S1：采用摄像头对驾驶图像进行采集；

S2：驾驶图像中定位驾驶员的眼睛区域，判断驾驶员是否佩戴墨镜，并获得粗定位图像；

S3：根据不同光线情况和驾驶员是否佩戴墨镜对粗定位图像进行处理得到面部轮廓图像；

S4：将面部轮廓图像归一化，传入预先训练好的EnlightenGAN模型进行光照增强，并转化得到灰度图像；

S5：从灰度图像中提取人脸中68个关键点的坐标，并将上述关键点坐标转换成numpy数组；

S6：从numpy数组中提取左右眼的点坐标并计算眼睛纵横比，当任一只眼睛的纵横比在一段时间中低于预先设定的阈值时，则发出利用有源蜂鸣器发出警报来提醒司机。

在步骤S3中，当光线较暗时或检测到驾驶员佩戴墨镜时，通过开启车载补光装置对驾驶员脸部进行补光以便于采集面部轮廓图像。

EnlightenGAN模型训练如下，

S41：获取同一场景下低光图像和正常光图像，并对低光图像和正常光图像的尺寸进行均一化处理，构成一个图像组；

S42：基于生成式对抗网络构建低光图像增强模型及损失函数，低光图像增强模型包括生成器网络和鉴别器网络，鉴别器网络包括全局鉴别器和局部鉴别器；

S43：训练低光图像增强模型：随机将处理后的数据集中的图像组输入到低光图像增强模型中进行训练，反复迭代通过最小化生成器网络或鉴别器网络的损失函数值来训练低光图像增强模型，直至低光图像增强模型达到纳什均衡状态，则训练完成；

S44：将待增强的低光图像输入到训练好的低光图像增强模型，获得增强后的正常光图像。

生成器网络包含8个卷积模块，每个卷积模块包含2个3x3的卷积层、LeakyReLu激活函数以及批归一化层。

全局鉴别器和局部鉴别器均采用了PatchGAN来用于真/假鉴别，除了图像级别的全局鉴别，还通过从输出和真实图像之间随机裁剪局部块来区分真/假以用作局部鉴别。