FaceRecognition(using_IPCAM).py

import urllib.request
import cv2
import numpy as np
import imutils
import os

haar_file = 'haarcascade_frontalface_default.xml'
face_cascade = cv2.CascadeClassifier(haar_file)

dataset = "Directory"
print('Training...')
(images, labels, names, id) = ([], [], {}, 0)

for (subdirs, dirs, files) in os.walk(dataset):
    for subdir in dirs:
        names[id] = subdir
        subjectpath = os.path.join(dataset, subdir)
        for filename in os.listdir(subjectpath):
            path = subjectpath + '/' + filename
            label = id
            images.append(cv2.imread(path, 0))
            labels.append(int(label))   
        id += 1    
(images,labels) = [np.array(lis) for lis in [images,labels]]
(width,height) = (130,100)
model= cv2.face.LBPHFaceRecognizer_create()
model.train(images, labels)
url='http://192.168.**.*:****/shot.jpg'
imgPath = urllib.request.urlopen(url)
cnt = 0
while True:
    imgPath = urllib.request.urlopen(url)
    imgNp = np.array(bytearray(imgPath.read()), dtype = np.uint8)
    img = cv2.imdecode(imgNp, -1)
    img2 = imutils.resize(img, width=450)
    gray = cv2.cvtColor(img2,cv2.COLOR_BGR2GRAY)
    faces = face_cascade.detectMultiScale(gray, 1.3, 4)
    for (x,y,w,h) in faces:
        cv2.rectangle(img2,(x,y),(x+w,y+h),(255,255,0),2)
        face = gray[y:y+h,x:x+w]
        face_resize = cv2.resize(face,(width,height))

        prediction = model.predict(face_resize)
        cv2.rectangle(img2,(x,y),(x+w,y+h),(0,255,0),3)
        if prediction[1]<800:
            cv2.putText(img2,'%s - %.0f' % (names[prediction[0]],prediction[1]),(x-10,y-10),cv2.FONT_HERSHEY_PLAIN,2,(0,0,255))
            print(names[prediction[0]])
            cnt = 0
        else:
            cnt += 1
            cv2.putText(img2,'Unknown',(x-10,y-10),cv2.FONT_HERSHEY_PLAIN,1,(0,255,0))
            if cnt > 100:
                print("Unknown Person")
                cv2.imwrite("Unknown.jpg",img2)
                cnt = 0
                
    cv2.imshow('FaceRecognition',img2)
    key = cv2.waitKey(10)
    if key == 27:
        break

cv2.destroyAllWindows()