78 lines
No EOL
3.8 KiB
Python
78 lines
No EOL
3.8 KiB
Python
import DBHelper
|
|
import shutil
|
|
from joblib import Parallel, delayed
|
|
import multiprocessing
|
|
import os
|
|
|
|
import Facial_Recognition_Enrollment
|
|
|
|
|
|
def remove_your_face(firstname, lastname, email, phone):
|
|
users = DBHelper.db.child("Users").get()
|
|
try:
|
|
count = 0
|
|
count2 = 0
|
|
for user in users.each():
|
|
count += 1
|
|
print("The specified user will be removed...")
|
|
for user in users.each():
|
|
count2 += 1
|
|
if DBHelper.get_firstname("User_" + str(count2)) == firstname and DBHelper.get_lastname(
|
|
"User_" + str(count2)) == lastname and DBHelper.get_email(
|
|
"User_" + str(count2)) == email and DBHelper.get_phone("User_" + str(count2)) == phone:
|
|
Parallel(n_jobs=multiprocessing.cpu_count())(
|
|
delayed(remove_parallel_user_photos)(i, count2) for i in range(10))
|
|
DBHelper.remove_data("User_" + str(count2))
|
|
shutil.rmtree("Facial_images/face_rec/train/User_" + str(count2))
|
|
print("Successfully removed the User.")
|
|
break
|
|
print("Reorganizing the Users...")
|
|
if count2 != count and count - count2 - 1 != 0:
|
|
for x in range(count - count2):
|
|
if not os.path.isdir("Facial_images/face_rec/train/User_" + str(count2)):
|
|
os.makedirs("Facial_images/face_rec/train/User_" + str(count2))
|
|
Parallel(n_jobs=multiprocessing.cpu_count())(
|
|
delayed(update_parallel_user_photos)(i, count2) for i in range(10))
|
|
DBHelper.upload_data("User_" + str(count2), DBHelper.get_firstname("User_" + str(count2 + 1)),
|
|
DBHelper.get_lastname("User_" + str(count2 + 1)),
|
|
DBHelper.get_email("User_" + str(count2 + 1)),
|
|
DBHelper.get_phone("User_" + str(count2 + 1)))
|
|
count2 += 1
|
|
DBHelper.remove_data("User_" + str(count))
|
|
shutil.rmtree("Facial_images/face_rec/train/User_" + str(count))
|
|
elif count2 != count and count - count2 - 1 == 0:
|
|
if not os.path.isdir("Facial_images/face_rec/train/User_" + str(count2)):
|
|
os.makedirs("Facial_images/face_rec/train/User_" + str(count2))
|
|
Parallel(n_jobs=multiprocessing.cpu_count())(
|
|
delayed(update_parallel_user_photos)(i, count2) for i in range(10))
|
|
DBHelper.upload_data("User_" + str(count2), DBHelper.get_firstname("User_" + str(count2 + 1)),
|
|
DBHelper.get_lastname("User_" + str(count2 + 1)),
|
|
DBHelper.get_email("User_" + str(count2 + 1)),
|
|
DBHelper.get_phone("User_" + str(count2 + 1)))
|
|
DBHelper.remove_data("User_" + str(count))
|
|
shutil.rmtree("Facial_images/face_rec/train/User_" + str(count))
|
|
print("Data saved! Starting enrollment...")
|
|
Facial_Recognition_Enrollment.enroll_face_dataset()
|
|
print("Face removal completed!")
|
|
print("Success.")
|
|
except:
|
|
print("No Users exist for User Removal.")
|
|
|
|
|
|
def remove_parallel_user_photos(i, count2):
|
|
DBHelper.delete_user_photo("User_" + str(count2) + "/" + str(i) + ".jpg")
|
|
|
|
|
|
def update_parallel_user_photos(i, count2):
|
|
DBHelper.download_user_photo_other("User_" + str(count2 + 1) + "/" + str(i) + ".jpg",
|
|
"User_" + str(count2) + "/" + str(i) + ".jpg")
|
|
DBHelper.upload_user_photo("User_" + str(count2) + "/" + str(i) + ".jpg")
|
|
DBHelper.delete_user_photo("User_" + str(count2 + 1) + "/" + str(i) + ".jpg")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
f = input('Enter your First Name:')
|
|
l = input('Enter your Last Name:')
|
|
e = input('Enter your E-Mail:')
|
|
p = input('Enter your Phone:')
|
|
remove_your_face(f, l, e, p) |