mirror of
https://github.com/serengil/deepface.git
synced 2025-06-06 19:45:21 +00:00
47 lines
1.2 KiB
Python
47 lines
1.2 KiB
Python
# 3rd party dependencies
|
|
import tensorflow as tf
|
|
|
|
# package dependencies
|
|
from deepface.commons import logger as log
|
|
|
|
logger = log.get_singletonish_logger()
|
|
|
|
|
|
def get_tf_major_version() -> int:
|
|
"""
|
|
Find tensorflow's major version
|
|
Returns
|
|
major_version (int)
|
|
"""
|
|
return int(tf.__version__.split(".", maxsplit=1)[0])
|
|
|
|
|
|
def get_tf_minor_version() -> int:
|
|
"""
|
|
Find tensorflow's minor version
|
|
Returns
|
|
minor_version (int)
|
|
"""
|
|
return int(tf.__version__.split(".", maxsplit=-1)[1])
|
|
|
|
|
|
def validate_for_keras3():
|
|
tf_major = get_tf_major_version()
|
|
tf_minor = get_tf_minor_version()
|
|
|
|
# tf_keras is a must dependency after tf 2.16
|
|
if tf_major == 1 or (tf_major == 2 and tf_minor < 16):
|
|
return
|
|
|
|
try:
|
|
import tf_keras
|
|
|
|
logger.debug(f"tf_keras is already available - {tf_keras.__version__}")
|
|
except ImportError as err:
|
|
# you may consider to install that package here
|
|
raise ValueError(
|
|
f"You have tensorflow {tf.__version__} and this requires "
|
|
"tf-keras package. Please run `pip install tf-keras` "
|
|
"or downgrade your tensorflow."
|
|
) from err
|