Updated the Integration. Tested the system, it works.

Facial_Recognition_Inference.py

@@ -4,33 +4,35 @@ import cv2
 import numpy as np
 import DBHelper
 
-try:
+
+def inference():
+    try:
         import cPickle  # Python 2
-except ImportError:
+    except ImportError:
         import _pickle as cPickle  # Python 3
 
-pwd = sys.path[0]
+    pwd = sys.path[0]
-PREDICTOR_PATH = pwd + '/Facial_models/shape_predictor_68_face_landmarks.dat'
+    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'
+    FACE_RECOGNITION_MODEL_PATH = pwd + '/Facial_models/dlib_face_recognition_resnet_model_v1.dat'
 
-SKIP_FRAMES = 1
+    SKIP_FRAMES = 1
-THRESHOLD = 0.4
+    THRESHOLD = 0.4
 
-faceDetector = dlib.get_frontal_face_detector()
+    faceDetector = dlib.get_frontal_face_detector()
-shapePredictor = dlib.shape_predictor(PREDICTOR_PATH)
+    shapePredictor = dlib.shape_predictor(PREDICTOR_PATH)
-faceRecognizer = dlib.face_recognition_model_v1(FACE_RECOGNITION_MODEL_PATH)
+    faceRecognizer = dlib.face_recognition_model_v1(FACE_RECOGNITION_MODEL_PATH)
 
-index = np.load(pwd + '/Facial_models/index.pkl', allow_pickle=True)
+    index = np.load(pwd + '/Facial_models/index.pkl', allow_pickle=True)
-faceDescriptorsEnrolled = np.load(pwd + '/Facial_models/descriptors.npy')
+    faceDescriptorsEnrolled = np.load(pwd + '/Facial_models/descriptors.npy')
 
-cam = cv2.VideoCapture(0)
+    cam = cv2.VideoCapture(0)
-count = 0
+    count = 0
 
-x1 = x2 = y1 = y2 = 0
+    x1 = x2 = y1 = y2 = 0
 
-cond = False
+    cond = False
 
-while DBHelper.get_power() == "on":
+    while DBHelper.get_power() == "on":
         t = time.time()
         success, im = cam.read()
 
@@ -97,6 +99,10 @@ while DBHelper.get_power() == "on":
             DBHelper.set_motor("off")
             DBHelper.set_alarm("on")
 
-DBHelper.set_alarm("off")
+    DBHelper.set_alarm("off")
-DBHelper.set_motor("off")
+    DBHelper.set_motor("off")
-cv2.destroyAllWindows()
+    cv2.destroyAllWindows()
+
+
+if __name__ == "__main__":
+    inference()

start_engine.py

@@ -32,7 +32,7 @@ def start():
         print("Success.")
     except:
         print("No Thieves are registered.")
-    Facial_Recognition_Inference
+    Facial_Recognition_Inference.inference()
 
 
 if __name__ == "__main__":