[OpenCV] 動画の顔認識

OpenCVのライブラリを使用して，動画での顔認識を行ってみた際のメモです．なお，Python3とOpenCVがインストールされている前提です（Python3のインストールはこちら，OpenCVのインストールはこちら）．

コードは以下のようになります．

import cv2 as cv

cap = cv.VideoCapture(0)

face_cascade = cv.CascadeClassifier('haarcascade_frontalface_default.xml')

eye_cascade = cv.CascadeClassifier('haarcascade_eye.xml')

while True:

    ret, frame = cap.read()

    gray = cv.cvtColor(frame, cv.COLOR_BGR2GRAY)

    faces = face_cascade.detectMultiScale(gray, 1.3, 5) # default setting

    print(len(faces))

    

    for (x, y, w, h) in faces:

        cv.rectangle(frame, (x, y), (x+w, y+h), (255, 0, 0), 2)

        eye_gray = gray[y:y+h, x:x+w]

        eye_color = frame[y:y+h, x:x+w]

        eyes = eye_cascade.detectMultiScale(eye_gray)

        for (ex, ey, ew, eh) in eyes:

            cv.rectangle(eye_color, (ex, ey), (ex+ew, ey+eh), (0, 255, 0), 2)

    cv.imshow('frame', frame)

    if cv.waitKey(1) & 0xFF == ord('q'):

        break

cv.destroyAllWindows()

以下は，コードの解説です．
1行目
import cv2 as cv
OpenCVのパッケージを読み込む．

3行目
変数 cap にカメラの画像を読み込むように設定する．0 はラップトップに付属のカメラを使用する場合．上手く認識できない場合は，0を1, 2などと変えてみると上手く行く場合が多い．