diff --git a/start_engine.py b/start_engine.py index 776b170a7..6869d05a7 100644 --- a/start_engine.py +++ b/start_engine.py @@ -7,6 +7,7 @@ def start(): # Downloads all the user and thief photos from database to the project folder first or updates them. count = 0 users = DBHelper.db.child("Users").get() + thieves = DBHelper.db.child("Thieves").get() try: for user in users.each(): count += 1 @@ -18,7 +19,7 @@ def start(): print("No Users are registered.") count = 0 try: - for user in users.each(): + for thief in thieves.each(): count += 1 if not os.path.isdir("Photos_of_Thieves/Thief_" + str(count)): os.makedirs("Photos_of_Thieves/Thief_" + str(count)) @@ -27,4 +28,6 @@ def start(): except: print("No Thieves for now.") Facial_Recognition_Wrapper.training_recognizer("Fisher") - Facial_Recognition_Wrapper.face_recognition_inference("Fisher") \ No newline at end of file + Facial_Recognition_Wrapper.face_recognition_inference("Fisher") + +start()