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

Binary file not shown.