FaceRecognition-python/main.py

from time import sleep
from picamera import PiCamera
import cv2
import numpy as np
import os
import serial

#ser = serial.Serial('/dev/ttyUSB0',9600)
def init():
	go = True
	#OpenShifter()
	#if Sensors():
	print('Initilizing Camera and cascade components..')
	try:
		camera = PiCamera()
		camera.resolution = (640, 480)
		sleep(2)
		print('[ OK ] Camera')
	except:
		print('[ Error ] Can not initialize PiCamera')
		go = False

	try:
		face_cascade = cv2.CascadeClassifier('haarcascade_frontalface_default.xml')
		sleep(1)
		print('[ OK ] CascadeClassifier')
	except:
		print('[ Error ] Can not load cascade File')
		go = False
	if (go):
		print('Starting Photo loop..')
		start(camera, face_cascade)
		
	else: 
		print('closing')
	#else:
		#print('Sensors not Satisfied')
	
def start( camera, face_cascade):
	j = 0
	while True:
	#try:
		camera.capture('img.jpg')			
		img = cv2.imread('img.jpg',1)
		gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)		
		faces = face_cascade.detectMultiScale(gray, 1.3, 5)			
		i = 0;
		for (x,y,w,h) in faces:
			i += 1
			j += 1 				
			roi_gray = gray[y:y+h, x:x+w]
			roi_color = img[y:y+h, x:x+w]
			#try:
				#cv2.imwrite('train/'+str(j)+'.jpg',img)
				#print(j)
			#except Exception as ex:
				#print(ex)
			#print('Drawing on face ', i)
			cv2.rectangle(img,(x,y),(x+w,y+h),(255,0,0),2)
			cv2.putText(img, recognize(roi_gray), (x, y), cv2.FONT_HERSHEY_PLAIN, 1.7, (255, 0, 0), 2)
		cv2.imshow('image',img) 
		cv2.waitKey(100)
		if cv2.waitKey(1) & 0xFF == ord('q'):
			cv2.destroyAllWindows()
			break
		
		#except:
			#print('[ Error ] Unexpected exception')
			#print('Closing')
			#break
			
def train(number):

	train_path = 'train/'
	face_recognizer = cv2.createLBPHFaceRecognizer()
	images = []
	labels = []
	image_names = os.listdir(train_path)
	for image_name in image_names:
		if image_name.startswith("."):
			continue;
		image_path = train_path + image_name
		image = cv2.imread(image_path)
		gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
		images.append(gray)
		labels.append(number)
	print('training')
	face_recognizer.train(images, np.array(labels))
	face_recognizer.save('faces/hesham')
	

def recognize(image):
	
	subjects = ['','Hesham','Saeed']
	face_recognizer = cv2.createLBPHFaceRecognizer()
	face_recognizer.load('faces/Hesham-Saeed')
	img = image.copy()	
	label= face_recognizer.predict(img)
	print(label)
	if label[0] == 1 and int(label[1]) < 200:
		return 'Hesham'
	elif label[0] == 2 and int(label[1]) < 200:
		return 'Saeed'
	else:
		return 'Unknown'
		
def Sensors():
		
	while 1:
		try:
			line = ser.readline()
			print(line)
			if 'A' in line:
				return True
			else:
				print('Sensors not OK..')

		except:
			print('[ Error ] Can not connect with arduino. Retrying..')
			
def OpenShifter():
	ser.write('O')

init()