Deepface Example

deepfaceExample.py

from flask import Flask, request, jsonify
from deepface import DeepFace
import matplotlib.pyplot as plt
import os
from minio import Minio

app = Flask(__name__)


# allow POST requests
@app.route('/faceSimilarity', methods=['POST'])
def faceCompare():

    data = request.get_json()
    img1 = data['img1']
    img2 = data['img2']
    result = verification(img1,img2)
    return jsonify(result)


if __name__ == '__main__':
    app.run()


def verification(img1, img2, max_threshold_to_verify=0.35, model_name="VGG-Face",
                 distance_metric="cosine"):

    output = {}

    # Connecting to minio server
    client = Minio('play.minio.io:9000',
                   access_key='Q3AM3UQ867SPQQA43P2F',
                   secret_key='zuf+tfteSlswRu7BJ86wekitnifILbZam1KYY3TG',
                   secure=True)
    # Downloading the images to VM
    client.fget_object("img", img1, img1)
    client.fget_object("img", img2, img2)
    print("Images downloaded!")
    result = DeepFace.verify(img1, img2, model_name=model_name,
                             distance_metric=distance_metric)
    output["verified"] = result["verified"]
    output["distance"] = result["distance"]
    output["similarity"] = (1 - float(result["distance"]))*100

    # Delete downloaded images
    if os.path.isfile(img1):
        os.remove(img1)
        os.remove(img2)
        print("Images are processed and deleted successfully!")
    return output