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

Update DeepFace.py

Browse Source 
build_model return type
This commit is contained in:
Sefik Ilkin Serengil 2023-12-18 11:56:56 +00:00 committed by GitHub
parent e91e48bd6c
commit 19f7cf620d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
deepface/DeepFace.py
View File

@ -43,10 +43,13 @@ os.environ["TF_CPP_MIN_LOG_LEVEL"] = "3"
tf_version = int(tf.__version__.split(".", maxsplit=1)[0])
if tf_version == 2:
tf.get_logger().setLevel(logging.ERROR)
from tensorflow.keras.models import Model
else:
from keras.models import Model
# -----------------------------------
def build_model(model_name: str) -> Any:
def build_model(model_name: str) -> Union[Model, Any]:
"""
This function builds a deepface model
Parameters:
@ -55,7 +58,7 @@ def build_model(model_name: str) -> Any:
Age, Gender, Emotion, Race for facial attributes
Returns:
built deepface model (keras.engine.Functional)
built deepface model ( (tf.)keras.models.Model )
"""
# singleton design pattern