FaceDetection.py

import cv2
import os
dataset="dataset"
name="face"
path=os.path.join(dataset, name)
if not os.path.isdir(path):
    os.mkdir(path)
(width,height)=(130,100)
alg= "haarcascade_frontalface_default.xml"
haar_cascade=cv2.CascadeClassifier(alg)
cam=cv2.VideoCapture(0)
count=0
text="No face Detected"
while count < 20:
    print(count)
    _,img=cam.read()
    grayimg=cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
    face=haar_cascade.detectMultiScale(grayimg,1.3,4)
    for(x,y,w,h) in face:
        cv2.rectangle(img,(x,y),(x+w,y+h),(0,255,0),2 )
        faceOnly=grayimg[y:y+h,x:x+w]
        resizeimg=cv2.resize(faceOnly,(width,height))
        cv2.imwrite("%s/%s.jpg" %(path,count),resizeimg)
        count+=1
        text="Face Detected"
    cv2.imshow("Facedetection",img)
    print(text)
    key=cv2.waitKey(10)
    if key==27:
        break
    print("Image captured successfully")
cam.release()
cv2.destroyAllWindows()