AddUnauthorizedAddNamesSinglePicture.py

from urllib.request import urlopen
import cv2
import os
import pickle
import numpy as np
import sqlite3
from PIL import Image
import datetime

path='SinglePicture'
num = 0

face_cascade = cv2.CascadeClassifier('Cascades/data/haarcascade_frontalface_default.xml')
rec = cv2.face.LBPHFaceRecognizer_create()
rec.read('Recognizer/trainningData.yml')
id=0
font=cv2.FONT_HERSHEY_SIMPLEX

def getProfile(id):
    conn=sqlite3.connect("facebase.db")
    cmd="SELECT * FROM People WHERE Id="+str(id)
    cursor=conn.execute(cmd)
    profile=None
    for row in cursor:
        profile=row
    conn.close()
    return profile

imagePaths=[os.path.join(path,f) for f in os.listdir(path)]
for imagePath in imagePaths:
    image=str(os.path.split(imagePath)[1])
img=cv2.imread('SinglePicture/'+image,1)
gray=cv2.imread('SinglePicture/'+image,0)
faces = face_cascade.detectMultiScale(gray, scaleFactor=1.3, minNeighbors=5)
for (x,y,w,h) in faces:
    id,conf=rec.predict(gray[y:y+h,x:x+w])
    profile=getProfile(id)
    if conf < 70:
        cv2.rectangle(img, (x,y), (x+w,y+h),(170,170,170),1)
        cv2.putText(img,str(profile[1]),(x+10,y+h+20),font, .65, (64,204,46),1)
    else:
        num = num+1
        cv2.rectangle(img, (x,y), (x+w,y+h),(27,133,255),1)
        cv2.putText(img,'Unauthorised'+'_'+str(num), (x,y+h+15),font, .55,(54,65,255),1)
cv2.imshow('View',img)
cv2.waitKey(0)
num = 0
for (x,y,w,h) in faces:
    id,conf=rec.predict(gray[y:y+h,x:x+w])
    profile=getProfile(id)
    if conf > 70:
        num = num+1
        id = str(input('Unauthorised'+'_'+str(num)+'_ id : '))
        name = str(input('Unauthorised'+'_'+str(num)+'_name : '))
        CT = datetime.datetime.now()
        cv2.imwrite("DataSet3/"+name+"_"+str(id)+"-"+str(CT.strftime("%Y-%m-%d-%H-%M-%S"))+".jpg",gray[y:y+h,x:x+w])

cv2.destroyAllWindows()