import cv2
import numpy as np
face_cascade = cv2.CascadeClassifier('haarcascade_frontalface_default.xml')
cap = cv2.VideoCapture(0)
Id = input('Enter ID: ')
sampleNum = 0
while True:
    ret, img = cap.read()
    gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)

    faces = face_cascade.detectMultiScale(gray, 1.3, 5)
    for (x,y,w,h) in faces:
        cv2.rectangle(img, (x,y), (x+w, y+h), (255, 0 ,0), 2)
        sampleNum += 1
        cv2.imwrite('dataSet/User.'+str(Id) +'.'+ str(sampleNum) + ".jpg", gray[y:y+h,x:x+w])
        cv2.waitKey(100)
    cv2.imshow('img', img)
    cv2.waitKey(1)
    if sampleNum>100:
        break;
cap.release()
cv2.destroyAllWindows()

    
        
