forked from Pragya9ps/Face-Recognition-Attendance-System
-
Notifications
You must be signed in to change notification settings - Fork 0
/
testing.py
29 lines (25 loc) · 966 Bytes
/
testing.py
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
import cv2
import numpy as np
recognizer = cv2.createLBPHFaceRecognizer()
recognizer.read('TrainingImageLabel/trainner.yml')
cascadePath = "haarcascade_frontalface_default.xml"
faceCascade = cv2.CascadeClassifier(cascadePath)
font = cv2.FONT_HERSHEY_SIMPLEX
cam = cv2.VideoCapture(0)
while True:
ret, im = cam.read()
gray = cv2.cvtColor(im, cv2.COLOR_BGR2GRAY)
faces = faceCascade.detectMultiScale(gray, 1.2, 5)
for(x, y, w, h) in faces:
Id, conf = recognizer.predict(gray[y:y+h, x:x+w])
# # else:
# # Id="Unknown"
# cv2.rectangle(im, (x-22,y-90), (x+w+22, y-22), (0,255,0), -1)
cv2.rectangle(im, (x, y), (x + w, y + h), (0, 260, 0), 7)
cv2.putText(im, str(Id), (x, y-40), font, 2, (255, 255, 255), 3)
# cv2.putText(im, str(Id), (x + h, y), font, 1, (0, 260, 0), 2)
cv2.imshow('im', im)
if cv2.waitKey(10) & 0xFF == ord('q'):
break
cam.release()
cv2.destroyAllWindows()