33 lines
1.2 KiB
Python
33 lines
1.2 KiB
Python
import os
|
|
import DBHelper
|
|
import Facial_Recognition_Wrapper
|
|
|
|
|
|
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
|
|
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")
|
|
except:
|
|
print("No Users are registered.")
|
|
count = 0
|
|
try:
|
|
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))
|
|
for i in range(20):
|
|
DBHelper.download_thief_photo("Thief_" + str(count) + "/" + str(i) + ".jpg")
|
|
except:
|
|
print("No Thieves for now.")
|
|
Facial_Recognition_Wrapper.training_recognizer("Fisher")
|
|
Facial_Recognition_Wrapper.face_recognition_inference("Fisher")
|
|
|
|
start()
|