Spaces:
Configuration error
Configuration error
| """ | |
| @author: Santhosh R | |
| """ | |
| import cv2,os | |
| import shutil | |
| import csv | |
| import numpy as np | |
| from PIL import Image, ImageTk | |
| import pandas as pd | |
| # This will make sure no duplicates exixts in profile.csv(using Pandas here) | |
| df = pd.read_csv('Profile.csv') | |
| df.sort_values('Ids', inplace = True) | |
| df.drop_duplicates(subset = 'Ids', keep = 'first', inplace = True) | |
| df.to_csv('Profile.csv', index = False) | |
| # Fuction to detect the face | |
| def DetectFace(): | |
| reader = csv.DictReader(open('Profile.csv')) | |
| print('Detecting Login Face') | |
| for rows in reader: | |
| result = dict(rows) | |
| #print(result) | |
| if result['Ids'] == '1': | |
| name1 = result['Name'] | |
| elif result['Ids'] == '2': | |
| name2 = result["Name"] | |
| recognizer = cv2.face.LBPHFaceRecognizer_create() #cv2.createLBPHFaceRecognizer() | |
| recognizer.read("TrainData\Trainner.yml") | |
| harcascadePath = "hh.xml" | |
| faceCascade = cv2.CascadeClassifier(harcascadePath); | |
| cam = cv2.VideoCapture(0) | |
| font = cv2.FONT_HERSHEY_SIMPLEX | |
| Face_Id = '' | |
| name2 = '' | |
| # Camera ON Everytime | |
| while True: | |
| ret, frame = cam.read() | |
| gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY) | |
| faces = faceCascade.detectMultiScale(gray, 1.3, 5) | |
| Face_Id = 'Not detected' | |
| # Drawing a rectagle around the face | |
| for (x, y, w, h) in faces: | |
| Face_Id = 'Not detected' | |
| cv2.rectangle(frame, (x, y), (x + w, y + h), (0, 255, 0), 3) | |
| Id, confidence = recognizer.predict(gray[y:y + h, x:x + w]) | |
| if (confidence < 80): | |
| if (Id == 1): | |
| name = name1 | |
| elif (Id == 2): | |
| name = name2 | |
| Predicted_name = str(name) | |
| Face_Id = Predicted_name | |
| else: | |
| Predicted_name = 'Unknown' | |
| Face_Id = Predicted_name | |
| # Here unknown faces detected will be stored | |
| noOfFile = len(os.listdir("UnknownFaces")) + 1 | |
| if int(noOfFile) < 100: | |
| cv2.imwrite("UnknownFaces\Image" + str(noOfFile) + ".jpg", frame[y:y + h, x:x + w]) | |
| else: | |
| pass | |
| cv2.putText(frame, str(Predicted_name), (x, y + h), font, 1, (255, 255, 255), 2) | |
| cv2.imshow('Picture', frame) | |
| #print(Face_Id) | |
| cv2.waitKey(1) | |
| # Checking if the face matches for Login | |
| if Face_Id == 'Not detected': | |
| print("-----Face Not Detected, Try again------") | |
| pass | |
| elif Face_Id == name1 or name2 and Face_Id != 'Unknown' : | |
| print('----------Detected as {}----------'.format(name1)) | |
| print('-----------login successfull-------') | |
| print('***********WELCOME {}**************'.format(name1)) | |
| break | |
| else: | |
| print('-----------Login failed please try agian-------') | |
| #if (cv2.waitKey(1) == ord('q')): | |
| # break | |
| DetectFace() | |