python-opencv人眼檢測

一、代碼：

# -*- coding: utf-8 -*-
"""
@Time    : 2018/3/5 14:32
@Author  : LITTLE HORSE
@Software: PyCharm
"""

import numpy as np
import cv2

# 導入訓練好的人臉以及人眼分類文件
face_cascade = cv2.CascadeClassifier('haarcascade_frontalface_default.xml')
eye_cascade = cv2.CascadeClassifier('haarcascade_eye.xml')

cap = cv2.VideoCapture(0)
cap.set(3, 320)
cap.set(4, 240)
while(cap.isOpened()):
    ret, img = cap.read()
    if ret == True:
        gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
        faces = face_cascade.detectMultiScale(gray, 1.3, 5)
        for (x ,y, w, h) in faces:
            img = cv2.rectangle(img, (x,y), (x+w,y+h), (255,0,0), 2)
            roi_gray = gray[y:y+h, x:x+w]
            roi_color = img[y:y+h, x:x+w]
            eyes = eye_cascade.detectMultiScale(roi_gray)
            for (ex, ey, ew, eh) in eyes:
                cv2.rectangle(roi_color, (ex, ey), (ex+ew, ey+eh), (0, 255, 0), 2)
        cv2.imshow('img', img)
    if cv2.waitKey(1) & 0xFF == ord('q'):
        break
cap.release()
cv2.destroyAllWindows()

二、分類器

用到的分類器可自行去opencv目錄查找。