"""Raspberry Pi Face Recognition Treasure Box

Treasure Box Script

Copyright 2013 Tony DiCola

"""

import cv2

import config

import face

# path to training images: training/positive

if __name__ == '__main__':

# Load training data into recognizer

print ('Loading training data...')

recognizer = cv2.face.createEigenFaceRecognizer()

recognizer.load(config.TRAINING_FILE)

print ('Training data loaded!')

# Initialize camera

camera = config.get_camera()

cv2.namedWindow('image', cv2.WINDOW_NORMAL)

print ('Running detection...')

print ('Press Ctrl-C to quit.')

while True:

# Check for the positive face and unlock if found.

image = camera.read()

# Convert image to grayscale.

image_gray = cv2.cvtColor(image, cv2.COLOR_RGB2GRAY)

# Get coordinates of single face in captured image.

result = face.detect_single(image_gray)

print (result)

if result is None:

cv2.imshow('image',image)

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

break

print ('Could not detect single face! Check the image in capture.pgm' \

' to see what was captured and try again with only one face visible.')

continue

x, y, w, h = result

cv2.rectangle(image, (x, y), (x+w, y+h), (0, 255, 0), 2)

cv2.imshow('image',image)

# Crop and resize image to face.

crop = face.resize(face.crop(image_gray, x, y, w, h))

# Test face against recognizer.

predicted = recognizer.predict(crop)

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

break

cv2.destroyAllWindows()