detectors.py

"""
:/helper class
"""

import cv2

class FaceDetector(object):
    
    def __init__(self):
        self.classifier = cv2.CascadeClassifier('/home/parth/Desktop/internal/xml/frontal_face.xml')

    def detect(self, image, biggest_only=True):
        is_color = len(image) == 3
        if is_color:
            image_gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
        else:
            image_gray = image
        scale_factor = 1.2

        min_neighbors = 5

        min_size = (30, 30)

        flags = cv2.CASCADE_FIND_BIGGEST_OBJECT | \
            cv2.CASCADE_DO_ROUGH_SEARCH if biggest_only else \
            cv2.CASCADE_SCALE_IMAGE

        face_coord = self.classifier.detectMultiScale(
            image_gray,
            scaleFactor=scale_factor,
            minNeighbors=min_neighbors,
            minSize=min_size,
            flags=flags
        )

        return face_coord