Fixed the face removal.

This commit is contained in:
Batuhan Berk Başoğlu 2021-02-10 10:09:59 -05:00
parent c654b33941
commit 68b0e75db2
4 changed files with 10 additions and 16 deletions

Binary file not shown.

Binary file not shown.

View file

@ -1,7 +1,5 @@
import DBHelper import DBHelper
import os
import shutil import shutil
import Facial_Recognition_Enrollment
def remove_your_face(firstname, lastname, email, phone): def remove_your_face(firstname, lastname, email, phone):
@ -24,36 +22,31 @@ def remove_your_face(firstname, lastname, email, phone):
break break
print("Reorganizing the Users... (This may take long time.)") print("Reorganizing the Users... (This may take long time.)")
if count2 != count and count - count2 - 1 != 0: if count2 != count and count - count2 - 1 != 0:
for x in range(count - count2 - 1): for x in range(count - count2):
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)))
if not os.path.isdir("Facial_images/face_rec/train/User_" + str(count2)):
os.makedirs("Facial_images/face_rec/train/User_" + str(count2))
for i in range(50): for i in range(50):
DBHelper.download_user_photo_other("User_" + str(count2 + 1) + "/" + str(i) + ".jpg", DBHelper.download_user_photo_other("User_" + str(count2 + 1) + "/" + str(i) + ".jpg",
"User_" + str(count2) + "/" + str(i) + ".jpg") "User_" + str(count2) + "/" + str(i) + ".jpg")
DBHelper.upload_user_photo("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") DBHelper.delete_user_photo("User_" + str(count2 + 1) + "/" + str(i) + ".jpg")
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 count2 += 1
DBHelper.remove_data("User_" + str(count)) DBHelper.remove_data("User_" + str(count))
shutil.rmtree("Facial_images/face_rec/train/User_" + str(count)) shutil.rmtree("Facial_images/face_rec/train/User_" + str(count))
elif count2 != count and count - count2 - 1 == 0: elif count2 != count and count - count2 - 1 == 0:
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)))
if not os.path.isdir("Facial_images/face_rec/train/User_" + str(count2)):
os.makedirs("Facial_images/face_rec/train/User_" + str(count2))
for i in range(50): for i in range(50):
DBHelper.download_user_photo_other("User_" + str(count2 + 1) + "/" + str(i) + ".jpg", DBHelper.download_user_photo_other("User_" + str(count2 + 1) + "/" + str(i) + ".jpg",
"User_" + str(count2) + "/" + str(i) + ".jpg") "User_" + str(count2) + "/" + str(i) + ".jpg")
DBHelper.upload_user_photo("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") DBHelper.delete_user_photo("User_" + str(count2 + 1) + "/" + str(i) + ".jpg")
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)) DBHelper.remove_data("User_" + str(count))
shutil.rmtree("Facial_images/face_rec/train/User_" + str(count)) shutil.rmtree("Facial_images/face_rec/train/User_" + str(count))
Facial_Recognition_Enrollment.enroll_face_dataset()
print("Success.") print("Success.")
except: except:
print("No Users exist for User Removal.") print("No Users exist for User Removal.")
@ -65,3 +58,4 @@ if __name__ == "__main__":
e = input('Enter your E-Mail:') e = input('Enter your E-Mail:')
p = input('Enter your Phone:') p = input('Enter your Phone:')
remove_your_face(f, l, e, p) remove_your_face(f, l, e, p)

Binary file not shown.