github/deepface
1
1
Fork 0
mirror of https://github.com/serengil/deepface.git synced 2025-06-06 19:45:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

tf2

Browse Source
This commit is contained in:
serengil 2020-12-13 20:43:29 +03:00
parent cdf86701e3
commit 863019f117
1 changed files with 8 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
tests/unit_tests.py
View File

@ -10,9 +10,14 @@ from deepface.commons import functions
import json
import time
#-----------------------------------------
import tensorflow as tf
import logging
tf.get_logger().setLevel(logging.ERROR)
tf_version = int(tf.__version__.split(".")[0])
if tf_version == 2:
import logging
tf.get_logger().setLevel(logging.ERROR)
#-----------------------------------------
@ -144,8 +149,7 @@ dataset = [
]
models = ['VGG-Face', 'Facenet', 'OpenFace', 'DeepFace', 'DeepID', 'Dlib']
#metrics = ['cosine', 'euclidean', 'euclidean_l2']
metrics = ['cosine']
metrics = ['cosine', 'euclidean', 'euclidean_l2']
passed_tests = 0; test_cases = 0