疲劳驾驶检测代码cpp

以下是一个简单的疲劳驾驶检测代码示例，使用 OpenCV 库进行图像处理：

#include <opencv2/opencv.hpp>
#include <iostream>

using namespace cv;
using namespace std;

int main()
{
    VideoCapture cap(0); // 打开默认摄像头
    if (!cap.isOpened())
    {
        cerr << "ERROR: Unable to open the camera" << endl;
        return 1;
    }

    // 初始化检测参数
    int blink_counter = 0;
    int yawn_counter = 0;
    int blink_threshold = 10;
    int yawn_threshold = 20;

    // 加载眼睛和嘴巴分类器
    CascadeClassifier eye_cascade, mouth_cascade;
    eye_cascade.load("haarcascade_eye.xml");
    mouth_cascade.load("haarcascade_mouth.xml");

    while (true)
    {
        Mat frame;
        cap >> frame;

        // 检测眼睛
        vector<Rect> eyes;
        eye_cascade.detectMultiScale(frame, eyes, 1.1, 2, 0 | CV_HAAR_SCALE_IMAGE, Size(30, 30));
        if (eyes.size() == 0)
        {
            blink_counter++;
        }
        else
        {
            blink_counter = 0;
        }

        // 检测嘴巴
        vector<Rect> mouths;
        mouth_cascade.detectMultiScale(frame, mouths, 1.1, 2, 0 | CV_HAAR_SCALE_IMAGE, Size(30, 30));
        if (mouths.size() == 0)
        {
            yawn_counter++;
        }
        else
        {
            yawn_counter = 0;
        }

        // 判断是否疲劳驾驶
        if (blink_counter > blink_threshold)
        {
            cout << "Driver is sleepy!" << endl;
            blink_counter = 0;
            yawn_counter = 0;
        }
        else if (yawn_counter > yawn_threshold)
        {
            cout << "Driver is tired!" << endl;
            blink_counter = 0;
            yawn_counter = 0;
        }

        // 显示图像
        imshow("Frame", frame);
        if (waitKey(1) == 27) // ESC键退出
        {
            break;
        }
    }

    cap.release();
    destroyAllWindows();
    return 0;
}

此代码使用 Haar 级联分类器检测眼睛和嘴巴，如果连续检测到一定数量的眨眼或张嘴，则判断驾驶员处于疲劳状态。您可以根据需要调整阈值和分类器。