43 lines
953 B
Python
43 lines
953 B
Python
import sys
|
|
import os
|
|
import math
|
|
import cv2
|
|
import Facial_Recognition_Enrollment
|
|
|
|
|
|
def register_your_face(label):
|
|
num_cap = 50
|
|
|
|
path = sys.path[0] + '/Facial_images/face_rec/train/' + label
|
|
|
|
folder = os.path.exists(path)
|
|
|
|
if not folder:
|
|
os.makedirs(path)
|
|
|
|
cap = cv2.VideoCapture(1)
|
|
c = 0
|
|
while c < num_cap:
|
|
ret, frame = cap.read()
|
|
|
|
cv2.imshow("capture", frame)
|
|
|
|
cv2.imwrite(path + '/' + str(c) + '.jpg', frame)
|
|
|
|
c = c + 1
|
|
cv2.waitKey(500)
|
|
|
|
cap.release()
|
|
cv2.destroyAllWindows()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
print("Face registration start...")
|
|
print()
|
|
label = input('Pleas enter your Name/Label: ')
|
|
register_your_face(label)
|
|
print("Data saved! Starting enrollment...")
|
|
print()
|
|
Facial_Recognition_Enrollment.enroll_face_dataset() # Need discuss and modify after intergrate with database.
|
|
print("Face registration completed!")
|
|
print()
|