Integrate face registration and enrollment

Facial_Recognition_Enrollment.py

@@ -9,6 +9,7 @@ try:
 except ImportError:
   import _pickle as cPickle  # Python3.
 
+def enroll_face_dataset():
   pwd = sys.path[0]
   PREDICTOR_PATH = pwd + '/Facial_models/shape_predictor_68_face_landmarks.dat'
   FACE_RECOGNITION_MODEL_PATH = pwd + '/Facial_models/dlib_face_recognition_resnet_model_v1.dat'

Facial_Recognition_Registration.py

@@ -2,10 +2,11 @@ import sys
 import os
 import math
 import cv2
+import Facial_Recognition_Enrollment
 
 
 def register_your_face(label):
-    num_cap = 60
+    num_cap = 50
 
     path = sys.path[0] + '/Facial_images/face_rec/train/' + label
 
@@ -14,7 +15,7 @@ def register_your_face(label):
     if not folder:
         os.makedirs(path)
 
-    cap = cv2.VideoCapture(0)
+    cap = cv2.VideoCapture(1)
     c = 0
     while c < num_cap:
         ret, frame = cap.read()
@@ -33,3 +34,4 @@ def register_your_face(label):
 if __name__ == "__main__":
     label = input('Enter a label: ')
     register_your_face(label)
+    Facial_Recognition_Enrollment.enroll_face_dataset()