opencvpython实战_OpenCV计算机视觉实战(Python版)

疲劳检测

#导入工具包

from scipy.spatial import distance as dist

from collections import OrderedDict

import numpy as np

import argparse

import time

import dlib

import cv2

FACIAL_LANDMARKS_68_IDXS = OrderedDict([

("mouth", (48, 68)),

("right_eyebrow", (17, 22)),

("left_eyebrow", (22, 27)),

("right_eye", (36, 42)),

("left_eye", (42, 48)),

("nose", (27, 36)),

("jaw", (0, 17))

])

# http://vision.fe.uni-lj.si/cvww2016/proceedings/papers/05.pdf

def eye_aspect_ratio(eye):

# 计算距离，竖直的

A = dist.euclidean(eye[1], eye[5])

B = dist.euclidean(eye[2], eye[4])

# 计算距离，水平的

C = dist.euclidean(eye[0], eye[3])

# ear值

ear = (A + B) / (2.0 * C)

return ear

# 输入参数

ap = argparse.ArgumentParser()

ap.add_argument("-p", "--shape-predictor", required=True,

help="path to facial landmark predictor")

ap.add_argument("-v", "--video", type=str, default="",

help="path to input video file")

args = vars(ap.parse_args())

# 设置判断参数

EYE_AR_THRESH = 0.3

EYE_AR_CONSEC_FRAMES = 3

# 初始化计数器

COUNTER = 0

TOTAL = 0

# 检测与定位工具

print("[INFO] loading facial landmark predictor...")

<