2020-11-14 20:25:24 +00:00
|
|
|
import os
|
2020-11-07 03:57:24 +00:00
|
|
|
import DBHelper
|
2020-11-14 20:25:24 +00:00
|
|
|
import Facial_Recognition_Wrapper
|
2020-11-07 03:57:24 +00:00
|
|
|
|
2020-11-07 02:59:40 +00:00
|
|
|
|
|
|
|
|
def start():
|
2020-11-07 03:57:24 +00:00
|
|
|
# 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()
|
|
|
|
|
try:
|
2020-11-15 16:49:49 +00:00
|
|
|
for user in users.each():
|
2020-11-15 01:43:56 +00:00
|
|
|
count += 1
|
2020-11-15 16:49:49 +00:00
|
|
|
if not os.path.isdir("Facial_images/face_rec/train/User_" + str(count)):
|
|
|
|
|
os.makedirs("Facial_images/face_rec/train/User_" + str(count))
|
|
|
|
|
for i in range(20):
|
|
|
|
|
DBHelper.download_user_photo("User_" + str(count) + "/" + str(i) + ".jpg")
|
2020-11-07 03:57:24 +00:00
|
|
|
except:
|
|
|
|
|
print("No Users are registered.")
|
|
|
|
|
count = 0
|
|
|
|
|
try:
|
2020-11-15 16:49:49 +00:00
|
|
|
for user in users.each():
|
2020-11-15 01:43:56 +00:00
|
|
|
count += 1
|
2020-11-15 16:49:49 +00:00
|
|
|
if not os.path.isdir("Photos_of_Thieves/Thief_" + str(count)):
|
|
|
|
|
os.makedirs("Photos_of_Thieves/Thief_" + str(count))
|
|
|
|
|
for i in range(20):
|
|
|
|
|
DBHelper.download_thief_photo("Thief_" + str(count) + "/" + str(i) + ".jpg")
|
2020-11-07 03:57:24 +00:00
|
|
|
except:
|
2020-11-14 20:25:24 +00:00
|
|
|
print("No Thieves for now.")
|
|
|
|
|
Facial_Recognition_Wrapper.training_recognizer("Fisher")
|
2020-11-15 16:50:50 +00:00
|
|
|
Facial_Recognition_Wrapper.face_recognition_inference("Fisher")
|
