github/deepface
1
1
Fork 0
mirror of https://github.com/serengil/deepface.git synced 2025-06-07 12:05:22 +00:00
Code Issues Packages Projects Releases Wiki Activity

issue 198

Browse Source
This commit is contained in:
Sefik Serengil 2021-03-10 21:36:41 +03:00
parent 289fe78195
commit 4137d55aac
1 changed files with 22 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
api/api.py
View File

@ -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 from flask import Flask, jsonify, request, make_response
import argparse import argparse
@ -6,9 +14,19 @@ import json
import time import time
from tqdm import tqdm from tqdm import tqdm
#------------------------------
import tensorflow as tf import tensorflow as tf
tf_version = int(tf.__version__.split(".")[0]) tf_version = int(tf.__version__.split(".")[0])
#------------------------------
if tf_version == 2:
import logging
tf.get_logger().setLevel(logging.ERROR)
#------------------------------
from deepface import DeepFace from deepface import DeepFace
#------------------------------ #------------------------------
@ -234,6 +252,10 @@ def verifyWrapper(req, trx_id = 0):
models["OpenFace"] = openface_model models["OpenFace"] = openface_model
models["DeepFace"] = deepface_model models["DeepFace"] = deepface_model
resp_obj = DeepFace.verify(instances, model_name = model_name, model = models) 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: else:
resp_obj = jsonify({'success': False, 'error': 'You must pass a valid model name. You passed %s' % (model_name)}), 205 resp_obj = jsonify({'success': False, 'error': 'You must pass a valid model name. You passed %s' % (model_name)}), 205