OpenCV-python 응용 <2. Smile recognition - 웃음 인식>

내 얼굴~

파란색으로 얼굴 인식! 노란색으로 웃음 인식! haarcascades를 이용한 얼굴과 웃음인식코드입니다.
haarcascades의 학습된 모델파일은 아래 OpenCV github에서 다운받을 수 있습니다. 

github.com/opencv/opencv/tree/master/data

opencv/opencv

사용 버전은 다음과 같습니다. 

• cv2 3.4.2
• numpy 1.18.1
• 파이썬 3.5.6

---

import numpy as np
import cv2

print("import clear")
faceCascade = cv2.CascadeClassifier('haarcascades/haarcascade_frontalface_default.xml')
smileCascade = cv2.CascadeClassifier('haarcascades/haarcascade_smile.xml')
print("recall model")
cap = cv2.VideoCapture(0)
cap.set(3,640) #set width
cap.set(4,480) #set Height
print("camera connected")

while True:
    ret, img = cap.read()
    gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
    faces = faceCascade.detectMultiScale(
        gray,
        scaleFactor=1.2,
        minNeighbors=5,
        minSize=(20, 20)
    )
    for (x,y,w,h) in faces:
        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]
        smile = smileCascade.detectMultiScale(
            roi_gray,
            scaleFactor = 1.5,
            minNeighbors=15,
            minSize=(25, 25),
        )
        if len(smile)==0:
           print("no smile")
        else:
            for (xx, yy, ww, hh) in smile:
                cv2.rectangle(roi_color, (xx, yy), (xx + ww, yy + hh), (0, 255, 255), 2)
        
        cv2.imshow('video', img)
    k= cv2.waitKey(30) & 0xff
    if k == 27:
        break 
    
cap.release()
cv2.destroyAllWindows()