dataCreator.py

import cv2
import numpy as np

faceDetector = cv2.CascadeClassifier('E:\Courses\Personnel\Own\Memo\MtotoSearch\FaceRecognitionFlask\haarcascade_frontalface_default.xml')
cam = cv2.VideoCapture(0)

id = input('Enter a id : ')
sampleNum = 0

while True:
    ret, img = cam.read()
    gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
    faces = faceDetector.detectMultiScale(gray, 1.3, 5)
    
    for(x,y,h,w) in faces:
        sampleNum = sampleNum+1
        cv2.imwrite('dataSet/train/1/ben.'+str(id)+'.'+str(sampleNum)+'.jpg', gray[y:y+h, x:x+w])
        cv2.rectangle(img, (x,y), (x+w, y+h), (0,0,255),2)
        cv2.waitKey(100)
    cv2.imshow("Face", img)
    cv2.waitKey(1)
    if(sampleNum>20):
        break
cam.release()
cv2.destroyAllWindows()