face_spoofing.py

import numpy as np
import io
import cv2
from sklearn.externals import joblib
from face_detector import get_face_detector, find_faces
from image_bytecode import image_bytecode



def calc_hist(img):
    histogram = [0] * 3
    for j in range(3):
        histr = cv2.calcHist([img], [j], None, [256], [0, 256])
        histr *= 255.0 / histr.max()
        histogram[j] = histr
    return np.array(histogram)


face_model = get_face_detector()
clf = joblib.load('models/face_spoofing.pkl')
sample_number = 1

measures = np.zeros(sample_number, dtype=np.float)


def spoof(image_obj):
    """
    :params image_obj: io.BytesIO instance of image
    :returns an io.BytesIO instance of image
    """
    count = 0
    flag = True
    while flag == True:
        # ret, img = cap.read()
        # face_capture=face_read()
        # path=img
        # img = image_bytecode(img)

        img = np.frombuffer(image_obj.getvalue(), np.uint8)
        img = cv2.imdecode(img, cv2.IMREAD_COLOR)

        # img should be of np array
        faces = find_faces(img, face_model)

        measures[count % sample_number] = 0
        height, width = img.shape[:2]
        for x, y, x1, y1 in faces:
            roi = img[y:y1, x:x1]
            point = (0, 0)

            img_ycrcb = cv2.cvtColor(roi, cv2.COLOR_BGR2YCR_CB)
            img_luv = cv2.cvtColor(roi, cv2.COLOR_BGR2LUV)

            ycrcb_hist = calc_hist(img_ycrcb)
            luv_hist = calc_hist(img_luv)

            feature_vector = np.append(ycrcb_hist.ravel(), luv_hist.ravel())
            feature_vector = feature_vector.reshape(1, len(feature_vector))
            prediction = clf.predict_proba(feature_vector)
            prob = prediction[0][1]
            measures[count % sample_number] = prob
            cv2.rectangle(img, (x, y), (x1, y1), (255, 0, 0), 2)
            point = (x, y-5)
            if 0 not in measures:
                text = "True"
                if np.mean(measures) >=0.79:

                    text = "False"
                    font = cv2.FONT_HERSHEY_SIMPLEX
                    cv2.putText(img=img, text=text, org=point, fontFace=font, fontScale=0.9, color=(0, 0, 255),
                                thickness=2, lineType=cv2.LINE_AA)
                    # return True

                else:
                    font = cv2.FONT_HERSHEY_SIMPLEX
                    cv2.putText(img=img, text=text, org=point, fontFace=font, fontScale=0.9,
                                color=(0, 255, 0), thickness=2, lineType=cv2.LINE_AA)
            flag = False

    count += 1
    # cv2.imwrite('images/flask_image.png', img=img)
    # out_image = cv2.imwrite(os.path.join(execution_path,  "flask"+filename), image)
    img = cv2.imencode('.png', img)[1]
    img_bytes = img.tobytes()
    img_obj = io.BytesIO(img_bytes)
    return img_obj

# if __name__=="__main__":
#     spoof()