face_authentication.py

# by @codassassin

import cv2
import numpy as np
import face_recognition
import os
import FaceMeshModule as fm


path = 'Auth_face'
images = []
classNames = []
myList = os.listdir(path)
for cl in myList:
    curImg = cv2.imread(f'{path}/{cl}')
    images.append(curImg)
    classNames.append(os.path.splitext(cl)[0])


def searchName(name):
    with open('encode.csv', 'r+') as f:
        myDataList = f.readlines()
        nameList = []
        for line in myDataList:
            entry = line.split(',')
            nameList.append(entry[0])
        if name in nameList:
            return True
        else:
            return False


def fetchEncoding(images):
    encodeList = []
    for img in images:
        img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
        encode = face_recognition.face_encodings(img)[0]
        encodeList.append(encode)
    return encodeList


def enterUser(name, registered):
    with open('encode.csv', 'r+') as f:
        myDataList = f.readlines()
        nameList = []
        for line in myDataList:
            entry = line.split(',')
            nameList.append(entry[0])
        if name not in nameList:
            f.writelines(f'\n{name},{registered}')


def confirmReg(name):
    with open('encode.csv', 'r') as f:
        myDataList = f.readlines()
        flag = 0
        for line in myDataList:
            entry = line.split(',')
            if entry[0] == name:
                flag = 1
                break
            else:
                flag = 0
        if flag == 1:
            return True
        else:
            return False


def compareEncodings(image):
    small_image = cv2.resize(image, (0, 0), None, 0.25, 0.25)
    small_image = cv2.cvtColor(small_image, cv2.COLOR_BGR2RGB)

    face = face_recognition.face_locations(small_image)
    encode = face_recognition.face_encodings(small_image, face)

    match = face_recognition.compare_faces(encodingComp, encode[0])
    face_dis = face_recognition.face_distance(encodingComp, encode[0])
    match_index = np.argmin(face_dis)

    if match[match_index]:
        print("The user is registered under another name")
        return True
    else:
        return False


encodingComp = fetchEncoding(images)
nameAuth = input('Enter your name: ')

if searchName(nameAuth):
    cap = cv2.VideoCapture(0)
    detector = fm.FaceMeshDetector()

    while True:
        success, img = cap.read()
        imag, faces = detector.findFaceMesh(img)
        imgS = cv2.resize(img, (0, 0), None, 0.25, 0.25)
        imgS = cv2.cvtColor(imgS, cv2.COLOR_BGR2RGB)

        facesCurFrame = face_recognition.face_locations(imgS)
        encodesCurFrame = face_recognition.face_encodings(imgS, facesCurFrame)

        flag = 1
        for encodeFace, faceLoc in zip(encodesCurFrame, facesCurFrame):
            matches = face_recognition.compare_faces(encodingComp, encodeFace)
            faceDis = face_recognition.face_distance(encodingComp, encodeFace)
            matchIndex = np.argmin(faceDis)
            if matches[matchIndex]:
                print("Identified as " + nameAuth)
                y1, x2, y2, x1 = faceLoc
                y1, x2, y2, x1 = y1 * 4, x2 * 4, y2 * 4, x1 * 4
                cv2.rectangle(img, (x1, y1), (x2, y2), (0, 255, 0), 2)
                cv2.rectangle(img, (x1, y2 - 35), (x2, y2), (0, 255, 0), cv2.FILLED)
                cv2.putText(img, 'Identity confirmed: ', (50, 30), cv2.FONT_HERSHEY_COMPLEX, 1, (255, 0, 0), 2)
                cv2.putText(img, nameAuth, (x1 + 6, y2 - 6), cv2.FONT_HERSHEY_COMPLEX, 1, (255, 255, 255), 2)
                flag = 1
            else:
                cv2.putText(img, 'Identity could not be confirmed! ', (50, 30), cv2.FONT_HERSHEY_COMPLEX, 1, (255, 0, 0), 2)
                flag = 0

        if flag == 0:
            break

        cv2.imshow('Webcam', img)
        cv2.waitKey(1)

else:
    cap = cv2.VideoCapture(0)

    while True:
        success, img = cap.read()
        cv2.putText(img, "Press C to capture your image", (50, 30), cv2.FONT_HERSHEY_COMPLEX, 1, (255, 0, 0), 1)
        cv2.imshow('Webcam', img)

        if cv2.waitKey(1) & 0xFF == ord('c'):
            break

    del cap
    if not classNames:
        cv2.imwrite('Auth_face/' + nameAuth + '.png', img)
        enterUser(nameAuth, True)
    else:
        if not compareEncodings(img):
            cv2.imwrite('Auth_face/' + nameAuth + '.png', img)
            enterUser(nameAuth, True)

    if confirmReg(nameAuth):
        print("User registered successfully !!")
    else:
        print("User could not be registered !!")