Formatting

README.md

@@ -1 +1 @@
-### Python code to run in Raspberry Pi with PiCamera to recognize certin faces and send a Serial signal 'o' to be received by arduino. 
+### Python code to run in Raspberry Pi with PiCamera to recognize certin faces.

main.py

@@ -64,7 +64,7 @@ def start( camera, face_cascade):
 		img = rawCapture.array
 		gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
 		faces = face_cascade.detectMultiScale(gray, 1.3,5)
-		i = 0;
+		i = 0
 		for (x,y,w,h) in faces:
 			i += 1
 			j += 1
@@ -110,6 +110,8 @@ def recognize(image,face_recognizer, names):
 		print(str(label) + ' >>'+names[label[0]])
 		return names[label[0]]
 
+#def DrawSquares(image,gray,faces,color):
+
 def OpenShifter():
 	try:
 		ser = serial.Serial('/dev/ttyUSB0',9600)

trainer.py

@@ -7,10 +7,10 @@ import os
 import sys
 import json
 
-
 picCount = 0
 new = False
 
+
 def start():
     print('Starting training..')
     label = int(input('Label (Integer): '))
@@ -32,22 +32,24 @@ def start():
     addPerson(label, newFileName, oldFileName)
     addName(name)
 
+
 def clearTrainFolder():
     print('clearing train folder')
-	filelist = [ f for f in os.listdir('train/') if f.endswith(".jpg") ]
+    filelist = [f for f in os.listdir('train/') if f.endswith(".jpg")]
     for f in filelist:
         os.remove(os.path.join('train/', f))
 
+
 def addPerson(label, newFileName, oldFileName):
     try:
-		#faceFilePath = '/faces/hesham-saeed2'
+        # faceFilePath = '/faces/hesham-saeed2'
-		faceFile = 'faces/'+ oldFileName
+        faceFile = 'faces/' + oldFileName
         face_recognizer = cv2.createLBPHFaceRecognizer()
         global new
         if new:
-			print 'new file. skipping load function'
+            print('new file. skipping load function')
         else:
-			print 'loading old file'
+            print('loading old file')
             face_recognizer.load(faceFile)
     except:
         print('[ Error ] Problem in init/loading LBPHfacerecognizer')
@@ -63,18 +65,19 @@ def addPerson(label, newFileName, oldFileName):
         gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
         face_cascade = cv2.CascadeClassifier('haarcascade_frontalface_default.xml')
         faces = face_cascade.detectMultiScale(gray, 1.3, 5)
-		for (x,y,w,h) in faces:
+        for (x, y, w, h) in faces:
-			images.append(gray[y:y+h, x:x+w])
+            images.append(gray[y:y + h, x:x + w])
             labels.append(label)
-			cv2.imshow('face',gray[y:y+h, x:x+w])
+            cv2.imshow('face', gray[y:y + h, x:x + w])
             cv2.waitKey(100)
-			print('Learning '+ image_name)
+            print('Learning ' + image_name)
     if new:
         face_recognizer.train(images, np.array(labels))
     else:
         face_recognizer.update(images, np.array(labels))
-	face_recognizer.save('faces/'+newFileName)
+    face_recognizer.save('faces/' + newFileName)
-	print('Updated and saved file in faces/'+newFileName) 
+    print('Updated and saved file in faces/' + newFileName)
+
 
 def capture(count):
     try:
@@ -86,17 +89,18 @@ def capture(count):
     except:
         print('[ Error ] Can not initialize PiCamera')
         sys.exit()
-	for i in range(1, count+1):
+    for i in range(1, count + 1):
-		pic = 'train/'+str(i)+'.jpg'
+        pic = 'train/' + str(i) + '.jpg'
         camera.capture(pic)
-		photo = cv2.imread(pic,1)
+        photo = cv2.imread(pic, 1)
-		cv2.imshow('Photo',photo)
+        cv2.imshow('Photo', photo)
-		print('Captured '+str(i)+'\nPath '+ pic)
+        print('Captured ' + str(i) + '\nPath ' + pic)
         cv2.waitKey(100)
         sleep(1)
     cv2.destroyAllWindows()
     print('Done!')
 
+
 def promptNew():
     filemode = raw_input('Make new File? (y/n): ')
     if filemode == 'y':
@@ -107,8 +111,8 @@ def promptNew():
         print('incorrect input')
         promptNew()
 
-def addName(newName):
 
+def addName(newName):
     with open("faces/names.json", "r") as read_file:
         exist = False
         namesJson = json.load(read_file)
@@ -117,12 +121,13 @@ def addName(newName):
         for name in names:
             if newName == name:
                 exist = True
-				print('name already in names list with index '+str(i))
+                print('name already in names list with index ' + str(i))
             i += 1
         if not exist:
             names.append(newName)
-			print('Added '+newName+'to names.json file')
+            print('Added ' + newName + 'to names.json file')
-	with open("faces/names.json","w") as write_file:
+    with open("faces/names.json", "w") as write_file:
-		json.dump(names,write_file)
+        json.dump(names, write_file)
+
 
 start()