From 4137d55aac89a521f1b166a62bcae30db1dbe2ae Mon Sep 17 00:00:00 2001 From: Sefik Serengil Date: Wed, 10 Mar 2021 21:36:41 +0300 Subject: [PATCH] issue 198 --- api/api.py | 22 ++++++++++++++++++++++ 1 file changed, 22 insertions(+) diff --git a/api/api.py b/api/api.py index 748869635..7fde7014d 100644 --- a/api/api.py +++ b/api/api.py @@ -1,3 +1,11 @@ +import warnings +warnings.filterwarnings("ignore") + +import os +os.environ['TF_CPP_MIN_LOG_LEVEL'] = '3' + +#------------------------------ + from flask import Flask, jsonify, request, make_response import argparse @@ -6,9 +14,19 @@ import time from tqdm import tqdm +#------------------------------ + import tensorflow as tf tf_version = int(tf.__version__.split(".")[0]) +#------------------------------ + +if tf_version == 2: + import logging + tf.get_logger().setLevel(logging.ERROR) + +#------------------------------ + from deepface import DeepFace #------------------------------ @@ -234,6 +252,10 @@ def verifyWrapper(req, trx_id = 0): models["OpenFace"] = openface_model models["DeepFace"] = deepface_model resp_obj = DeepFace.verify(instances, model_name = model_name, model = models) + + for key in resp_obj: #issue 198. + resp_obj[key]['verified'] = bool(resp_obj[key]['verified']) + else: resp_obj = jsonify({'success': False, 'error': 'You must pass a valid model name. You passed %s' % (model_name)}), 205