Count faces

faces.py

from cv2 import imread, cvtColor, CascadeClassifier, COLOR_BGR2GRAY, CASCADE_SCALE_IMAGE

HAARCASCADE_FRONTALFACE = '{base_path}/cascades/haarcascade_frontalface_default.xml'.format(
    base_path=os.path.abspath(os.path.dirname(__file__).replace('configs', ''))
)      

num_faces = 0
locations = []

image = imread ('PATH/TO/IMAGE')
gray = cvtColor(image, COLOR_BGR2GRAY)
detector = CascadeClassifier(HAARCASCADE_FRONTALFACE)

rects = detector.detectMultiScale(
    gray,
    scaleFactor=1.1,
    minNeighbors=5,
    minSize=(30, 30),
    flags=CASCADE_SCALE_IMAGE
)
locations = [(int(x), int(y), int(x + w), int(y + h)) for (x, y, w, h) in rects]
num_faces = len(rects)

return {'success': True, 'num_faces': num_faces, 'locations': locations}