import cv2
import numpy as np

face_cascade = cv2.CascadeClassifier('haarcascade_frontalface_default.xml')

if __name__ == "__main__":

    img = cv2.imread('TestFaces/will-smith.jpg')

    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)


    cv2.imshow('img', img)

    crop_img = img[y:y+h, x:x+w]
    cv2.imshow("cropped", crop_img)

    cv2.imwrite("test-crop.jpg",crop_img)

    cv2.waitKey(0)
    cv2.destroyAllWindows()