facerecognition/python3 at master · m-samik/facerecognition · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77


import cv2
import numpy as np
import webbrowser
import os

face_classifier =
cv2.CascadeClassifier('haarcascade_frontalface_default.xml')

def face_detector(img, size=0.5):

     # Convert image to grayscale
     gray = cv2.cvtColor(img,cv2.COLOR_BGR2GRAY)
     faces = face_classifier.detectMultiScale(gray, 1.3, 5)
     if faces is ():
         return img, []

     for (x,y,w,h) in faces:
         cv2.rectangle(img,(x,y),(x+w,y+h),(0,255,255),2)
         roi = img[y:y+h, x:x+w]
         roi = cv2.resize(roi, (200, 200))
     return img, roi


# Open Webcam
cap = cv2.VideoCapture(0)

while True:

     ret, frame = cap.read()

     image, face = face_detector(frame)

     try:
         face = cv2.cvtColor(face, cv2.COLOR_BGR2GRAY)

         # Pass face to prediction model
         # "results" comprises of a tuple containing the label and the
confidence value
         results = sami_model.predict(face)
         print(results)
         if results[1] < 500:
             confidence = int( 100 * (1 - (results[1])/400) )
             display_string = str(confidence) + '% Confident it is User'

         cv2.putText(image, display_string, (100, 120),
cv2.FONT_HERSHEY_COMPLEX, 1, (255,120,150), 2)

         if confidence > 87:
             os.system ("usermod -U sami")
             cv2.putText(image, "Hey Sami", (250, 450),
cv2.FONT_HERSHEY_COMPLEX, 1, (0,255,0), 2)
             cv2.imshow('Face Recognition', image )

         else:
             cv2.putText(image, "Wrong Face", (250, 450),
cv2.FONT_HERSHEY_COMPLEX, 1, (0,0,255), 2)
             cv2.imshow('Face Recognition', image )

     except:
         cv2.putText(image, "No Face Found", (220, 120) ,
cv2.FONT_HERSHEY_COMPLEX, 1, (0,0,255), 2)
         cv2.putText(image, "Locked", (250, 450),
cv2.FONT_HERSHEY_COMPLEX, 1, (0,0,255), 2)
         cv2.imshow('Face Recognition', image )
         pass

     if cv2.waitKey(1) == 13: #13 is the Enter Key
         break

cap.release()
cv2.destroyAllWindows()

cap.release()
cv2.destroyAllWindows()
print("Collecting Samples Complete")