# This script performs face recognition and draws a green box around a detected face.
import cv2
import sys

# The argument this script takes is a haarcascade.xml file which contains face parameters to be recognized.
cascPath = sys.argv[1]
faceCascade = cv2.CascadeClassifier(cascPath)

# Start the capture
video_capture = cv2.VideoCapture(0)

count = 10

while True:
    # Capture frame-by-frame
    ret, frame = video_capture.read()
    if not ret: continue
    # Analyze 1 out of 40 frames
    if(count % 40 == 0):
      gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)

      faces = faceCascade.detectMultiScale(
          gray,
          scaleFactor=1.1,
          minNeighbors=5,
          minSize=(10, 10),
          #flags=cv2.cv.CV_HAAR_SCALE_IMAGE
      )

      # Draw a rectangle around the faces
      for (x, y, w, h) in faces:
          cv2.rectangle(frame, (x, y), (x+w, y+h), (0, 255, 0), 2)

      # Display the resulting frame
      cv2.imshow('Video', frame)

      if cv2.waitKey(1) & 0xFF == ord('q'):
          break
    count += 1

# When everything is done, release the capture
video_capture.release()
cv2.destroyAllWindows()
