You are on page 1of 4

import cv2

import selenium

import numpy as np

from os import listdir

from os.path import join

from selenium import webdriver

from selenium.webdriver.common.keys import Keys

from selenium.webdriver.common.by import By

from selenium.webdriver.support.ui import WebDriverWait

from selenium.webdriver.support import expected_conditions as EC

def face_detect(img):

gray=cv2.cvtColor(img,cv2.COLOR_BGR2GRAY)

gray=cv2.resize(gray,(512,512))

faces=face_recog.detectMultiScale(gray,1.1,4)

if len(faces)==0:

return img,[],100,200,0,0

for x,y,w,h in faces:

cv2.rectangle(img,(x,y),(x+w,y+h),(0,255,0),1)

roi=img[y:y+h,x:x+w]

roi=cv2.resize(roi,(512,512))

return img,roi,x,y,w,h

def detect(photo):

try:
photo=cv2.cvtColor(photo,cv2.COLOR_BGR2GRAY)

result=model.predict(photo)

if result[1]<500:

con=100*(1-(result[1]/300))

if con>80:

output=d[result[0]]

else:

output="unknown"

except:

output="face not found"

finally:

return output

def facebook():

driver = webdriver.Chrome(

executable_path=r"C:\Users\hp\PycharmProjects\untitled\venv\Lib\site-packages\
chromedriver_win32\chromedriver.exe")

driver.get("https://www.facebook.com")

element = driver.find_element_by_id('email')

element.send_keys("XXXXXX")

element = driver.find_element_by_id('pass')

element.send_keys("XXXXXXX")

element.send_keys(Keys.RETURN)

data_path=r"C:/Users/hp/PycharmProjects/images/"
d={0:'saikiran',1:'gopi',2:'varun',3:'sravya'}

training=[]

label=[]

for i in listdir(data_path):

x=join(data_path,i)

for j in listdir(x):

image_path=join(x,j)

image=cv2.imread(image_path,cv2.IMREAD_GRAYSCALE)

training.append(np.asarray(image,dtype=np.uint8))

label.append(int(i))

training=np.asarray(training)

label=np.asarray(label)

model=cv2.face.LBPHFaceRecognizer_create()

model.train(training,np.asarray(label))

print("trained")

cap=cv2.VideoCapture(0)

com=1

face_recog=cv2.CascadeClassifier("haarcascade_frontalface_default.xml")

while(1):

ret,frame=cap.read()

frame=cv2.resize(frame,(512,512))

image,face,x,y,w,h=face_detect(frame)

font=cv2.FONT_HERSHEY_COMPLEX

text=detect(face)
cv2.putText(frame,text,(x,y),font,2,(0,255,255),1)

cv2.imshow('image',frame)

if text=="saikiran" and com==0:

txt="Do you want to login FB??"

cv2.putText(frame, txt, (0,100), font, 0.5, (0, 255, 255), 1)

cv2.imshow('image',frame)

key=cv2.waitKey(5000)

if key==ord('f'):

facebook()

com=1

else:

com=1

k=cv2.waitKey(1)

if k==27:

break

elif k==ord('a'):

com=0

cap.release()

cv2.destroyAllWindows()

You might also like