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,37 +31,39 @@ 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
 	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 = i + 1
-				j = 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)
-				
-				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
+	#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')
@@ -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()
-
-