疲劳驾驶检测系统:代码实战(基于OpenCV和Dlib,Python 3)

环境安装与配置(Windows 10)

步骤一:安装以下Python库

scipy、OpenCV、numpy、imutils、pyglet

建议用Anaconda 3。上面的库都可以用pip install XXX来安装,OpenCV的安装教程参考:Install OpenCV3 on Windows

步骤二:安装Dlib(Python 3)

在安装Dlib之前,需要确保已经安装好Visual Studio 2015(update 3)和CMake。

这里附上资源下载链接:

Visual Studio 2015:https://pan.baidu.com/s/1kF3gQv3C0OKwIPAXcB-lvQ;提取码:9e8n

VS2015 update 3:https://pan.baidu.com/s/1KdpGtIsTB4x3WaiKHhib9Q;提取码:vh5e

备注:在VS2015 update 3中安装C++编译模块。

CMake安装:pip install cmake==3.8.2

下载Dlib-19.6压缩包:http://dlib.net/files/dlib-19.6.zip

cd dlib-19.6\
mkdir build
cd build

# This is a single command. Backticks are used for line continuation
cmake -G "Visual Studio 14 2015 Win64" `
-DJPEG_INCLUDE_DIR=..\dlib\external\libjpeg `
-DJPEG_LIBRARY=..\dlib\external\libjpeg `
-DPNG_PNG_INCLUDE_DIR=..\dlib\external\libpng `
-DPNG_LIBRARY_RELEASE=..\dlib\external\libpng `
-DZLIB_INCLUDE_DIR=..\dlib\external\zlib `
-DZLIB_LIBRARY_RELEASE=..\dlib\external\zlib `
-DCMAKE_INSTALL_PREFIX=install ..

cmake --build . --config Release --target INSTALL
cd ..

或者通过pip install dlib==19.16安装:

 

代码资源下载与运行

Github开源项目:driver-fatigue-detection-system

1. git clone https://github.com/raja434/driver-fatigue-detection-system.git

2. 下载预训练好的脸部标志检测器:dlib’s pre-trained facial landmark detector。把下载好的数据放在项目文件夹中(与alarm.wav同一目录下)。

3. 运行项目:python drowsiness detection.py

实验运行结果展示

   

左图为清醒-睁眼状态,EAR值为0.33;右图为疲劳-眯眼状态,EAR值为0.19,促发警报“Drowsiness Alert”并发出警报声音。

原理解析

这个简单的疲劳监测系统是通过计算眼睛纵横比(Eye Aspect Ratio)来估计疲劳程度的,源码如下:

def eye_aspect_ratio(eye):
	# compute the euclidean distances between the two sets of
	# vertical eye landmarks (x, y)-coordinates
	A = dist.euclidean(eye[1], eye[5])
	B = dist.euclidean(eye[2], eye[4])

	# compute the euclidean distance between the horizon
	# eye landmark (x, y)-coordinates
	C = dist.euclidean(eye[0], eye[3])

	# compute the eye aspect ratio
	ear = (A + B) / (2.0 * C)

	# return the eye aspect ratio
	return ear

代码中eye[0] -> eye[5]就是下图中的P1 -> P6,纵横比就是“眼睛宽度 / 眼睛高度”,这个也很好理解。

下面设置预警的阈值为0.3,即EAR < 0.3时表示眨眼(blink)了一次或者处于闭眼状态,当闭眼帧数超过48时,触发警报。

# define two constants, one for the eye aspect ratio to indicate
# blink and then a second constant for the number of consecutive
# frames the eye must be below the threshold for to set off the
# alarm
EYE_AR_THRESH = 0.3
EYE_AR_CONSEC_FRAMES = 48

下面代码中把左眼和右眼的EAR值做平均,得到最终的EAR值。

# extract the left and right eye coordinates, then use the
# coordinates to compute the eye aspect ratio for both eyes
leftEye = shape[lStart:lEnd]
rightEye = shape[rStart:rEnd]
leftEAR = eye_aspect_ratio(leftEye)
rightEAR = eye_aspect_ratio(rightEye)

# average the eye aspect ratio together for both eyes
ear = (leftEAR + rightEAR) / 2.0

参考资料

1. Github开源项目:https://github.com/raja434/driver-fatigue-detection-system

2. Learn OpenCV:Install Dlib on Windows

发布了9 篇原创文章 · 获赞 25 · 访问量 4824

猜你喜欢

转载自blog.csdn.net/Carson1145/article/details/105160516