Updated Upload_Face.

Facial_Recognition_Enrollment.py

@@ -73,3 +73,7 @@ def enroll_face_dataset():
     np.save(pwd + '/Facial_models/descriptors.npy', faceDescriptors)
     with open(pwd + '/Facial_models/index.pkl', 'wb') as f:
         cPickle.dump(index, f)
+
+
+if __name__ == "__main__":
+    enroll_face_dataset()

Facial_Recognition_Inference.py

@@ -4,7 +4,6 @@ import cv2
 import numpy as np
 import DBHelper
 
-
 def inference():
     try:
         import cPickle  # Python 2
@@ -103,6 +102,5 @@ def inference():
     DBHelper.set_motor("off")
     cv2.destroyAllWindows()
 
-
 if __name__ == "__main__":
-    inference()
+    inference()

Facial_Recognition_Registration.py

@@ -15,7 +15,7 @@ def register_your_face(label):
     if not folder:
         os.makedirs(path)
 
-    cap = cv2.VideoCapture(1)
+    cap = cv2.VideoCapture(0)
     c = 0
     while c < num_cap:
         ret, frame = cap.read()

Upload_Face.py

@@ -13,8 +13,11 @@ def upload_your_face(firstname, lastname, email, phone):
         for user in users.each():
             count += 1
         DBHelper.upload_data("User_" + str(count), firstname, lastname, email, phone)
+        print("Face registration start...")
         Facial_Recognition_Registration.register_your_face("User_" + str(count))
+        print("Data saved! Starting enrollment...")
         Facial_Recognition_Enrollment.enroll_face_dataset()
+        print("Face registration completed!")
         for i in range(20):
             DBHelper.upload_user_photo("User_" + str(count) + "/" + str(i) + ".jpg")
     except: