在 Python 中使用 cv2 进行嗜睡检测

最新推荐文章于 2024-07-02 16:38:00 发布

原创

最新推荐文章于 2024-07-02 16:38:00 发布 · 429 阅读

4 ·

CC 4.0 BY-SA版权

文章标签：

#python #pygame #开发语言

本文介绍如何使用 Python 的 cv2 和 dlib 库构建一个简单的疲劳检测应用，通过检测视频中的人物眼睛状态来判断是否处于嗜睡状态。项目代码简洁，不到80行，包括了图像处理、面部检测和眼点识别等步骤。当检测到连续几帧的眼睛闭合超过预设阈值时，程序会触发警告。文章还提到了其他疲劳检测方法，如训练神经网络。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

大家好，在这个博客中，我们将构建一个嗜睡检测应用程序，它将检测视频中的人是否变得昏昏欲睡。

这是一个非常有趣且简单的项目，代码甚至不到 80 行，让我们开始吧

看看最终输出

注意——你不会在这里听到警报声，因为它是 GIF，你可以在博客上查看视频：

https://machinelearningprojects.net/drowsiness-detection/

嗜睡检测代码

from imutils import face_utils
import dlib
import cv2
from pygame import mixer

thres = 6
mixer.init()
sound = mixer.Sound('alarm.wav')
dlist = []
detector = dlib.get_frontal_face_detector()
predictor = dlib.shape_predictor("shape_predictor_68_face_landmarks.dat")
cap = cv2.VideoCapture(0)

def dist(a,b):
    x1,y1 = a
    x2,y2 = b
    return ((x1-x2)**2 + (y1-y2)**2)**0.5
 
while True:
    # Getting out image by webcam 
    _, image = cap.read()
    # Converting the image to gray scale
    gr