Working with Hesham-Saeed

main.py

@@ -1,13 +1,16 @@
 
 from time import sleep
 from picamera import PiCamera
-from io import BytesIO
 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()
@@ -28,8 +31,11 @@ def init():
 	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
@@ -41,17 +47,16 @@ def start( camera, face_cascade):
 		faces = face_cascade.detectMultiScale(gray, 1.3, 5)			
 		i = 0;
 		for (x,y,w,h) in faces:
-				i = i + 1
-				j = j + 1 				
-
+			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',roi_color)
-				except Exception as ex:
-					print(ex)
-				print('Drawing on face ', i)
-				
+			#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) 
@@ -87,17 +92,34 @@ def train(number):
 
 def recognize(image):
 	
-	subjects = ['','Hesham','Unknown']
+	subjects = ['','Hesham','Saeed']
 	face_recognizer = cv2.createLBPHFaceRecognizer()
-	face_recognizer.load('faces/hesham')
+	face_recognizer.load('faces/Hesham-Saeed')
 	img = image.copy()	
 	label= face_recognizer.predict(img)
-	if label[0] == 1:
+	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()
-
-