common function to get tf major version

deepface/DeepFace.py

@@ -29,7 +29,7 @@
 
 warnings.filterwarnings("ignore")
 os.environ["TF_CPP_MIN_LOG_LEVEL"] = "3"
-tf_version = int(tf.__version__.split(".", maxsplit=1)[0])
+tf_version = functions.get_tf_major_version()
 if tf_version == 2:
     tf.get_logger().setLevel(logging.ERROR)
 # -----------------------------------

deepface/basemodels/ArcFace.py

@@ -1,6 +1,5 @@
 import os
 import gdown
-import tensorflow as tf
 from deepface.commons import functions
 from deepface.commons.logger import Logger
 from deepface.models.FacialRecognition import FacialRecognition
@@ -12,7 +11,7 @@
 # --------------------------------
 # dependency configuration
 
-tf_version = int(tf.__version__.split(".", maxsplit=1)[0])
+tf_version = functions.get_tf_major_version()
 
 if tf_version == 1:
     from keras.models import Model

deepface/basemodels/DeepID.py

@@ -1,13 +1,12 @@
 import os
 import gdown
-import tensorflow as tf
 from deepface.commons import functions
 from deepface.commons.logger import Logger
 from deepface.models.FacialRecognition import FacialRecognition
 
 logger = Logger(module="basemodels.DeepID")
 
-tf_version = int(tf.__version__.split(".", maxsplit=1)[0])
+tf_version = functions.get_tf_major_version()
 
 if tf_version == 1:
     from keras.models import Model

deepface/basemodels/Facenet.py

@@ -1,6 +1,5 @@
 import os
 import gdown
-import tensorflow as tf
 from deepface.commons import functions
 from deepface.commons.logger import Logger
 from deepface.models.FacialRecognition import FacialRecognition
@@ -10,7 +9,7 @@
 # --------------------------------
 # dependency configuration
 
-tf_version = int(tf.__version__.split(".", maxsplit=1)[0])
+tf_version = functions.get_tf_major_version()
 
 if tf_version == 1:
     from keras.models import Model

deepface/basemodels/FbDeepFace.py

@@ -1,7 +1,6 @@
 import os
 import zipfile
 import gdown
-import tensorflow as tf
 from deepface.commons import functions
 from deepface.commons.logger import Logger
 from deepface.models.FacialRecognition import FacialRecognition
@@ -11,7 +10,7 @@
 # --------------------------------
 # dependency configuration
 
-tf_version = int(tf.__version__.split(".", maxsplit=1)[0])
+tf_version = functions.get_tf_major_version()
 
 if tf_version == 1:
     from keras.models import Model, Sequential

deepface/basemodels/OpenFace.py

@@ -7,7 +7,7 @@
 
 logger = Logger(module="basemodels.OpenFace")
 
-tf_version = int(tf.__version__.split(".", maxsplit=1)[0])
+tf_version = functions.get_tf_major_version()
 if tf_version == 1:
     from keras.models import Model
     from keras.layers import Conv2D, ZeroPadding2D, Input, concatenate
@@ -30,6 +30,7 @@ class OpenFace(FacialRecognition):
     """
     OpenFace model class
     """
+
     def __init__(self):
         self.model = load_model()
         self.model_name = "OpenFace"

deepface/basemodels/VGGFace.py

@@ -1,6 +1,5 @@
 import os
 import gdown
-import tensorflow as tf
 from deepface.commons import functions
 from deepface.commons.logger import Logger
 from deepface.models.FacialRecognition import FacialRecognition
@@ -9,8 +8,7 @@
 
 # ---------------------------------------
 
-tf_version = int(tf.__version__.split(".", maxsplit=1)[0])
-
+tf_version = functions.get_tf_major_version()
 if tf_version == 1:
     from keras.models import Model, Sequential
     from keras.layers import (

deepface/commons/functions.py

@@ -21,9 +21,12 @@
 # --------------------------------------------------
 # configurations of dependencies
 
-tf_version = tf.__version__
-tf_major_version = int(tf_version.split(".", maxsplit=1)[0])
-tf_minor_version = int(tf_version.split(".")[1])
+
+def get_tf_major_version() -> int:
+    return int(tf.__version__.split(".", maxsplit=1)[0])
+
+
+tf_major_version = get_tf_major_version()
 
 if tf_major_version == 1:
     from keras.preprocessing import image

deepface/extendedmodels/Age.py

@@ -1,7 +1,6 @@
 import os
 import gdown
 import numpy as np
-import tensorflow as tf
 from deepface.basemodels import VGGFace
 from deepface.commons import functions
 from deepface.commons.logger import Logger
@@ -12,7 +11,7 @@
 # ----------------------------------------
 # dependency configurations
 
-tf_version = int(tf.__version__.split(".", maxsplit=1)[0])
+tf_version = functions.get_tf_major_version()
 
 if tf_version == 1:
     from keras.models import Model, Sequential

deepface/extendedmodels/Emotion.py

@@ -1,6 +1,5 @@
 import os
 import gdown
-import tensorflow as tf
 import numpy as np
 import cv2
 from deepface.commons import functions
@@ -13,7 +12,7 @@
 # pylint: disable=line-too-long
 # -------------------------------------------
 # dependency configuration
-tf_version = int(tf.__version__.split(".", maxsplit=1)[0])
+tf_version = functions.get_tf_major_version()
 
 if tf_version == 1:
     from keras.models import Sequential

deepface/extendedmodels/Gender.py

@@ -1,6 +1,5 @@
 import os
 import gdown
-import tensorflow as tf
 import numpy as np
 from deepface.basemodels import VGGFace
 from deepface.commons import functions
@@ -14,8 +13,7 @@
 # -------------------------------------
 # dependency configurations
 
-tf_version = int(tf.__version__.split(".", maxsplit=1)[0])
-
+tf_version = functions.get_tf_major_version()
 if tf_version == 1:
     from keras.models import Model, Sequential
     from keras.layers import Convolution2D, Flatten, Activation

deepface/extendedmodels/Race.py

@@ -1,6 +1,5 @@
 import os
 import gdown
-import tensorflow as tf
 import numpy as np
 from deepface.basemodels import VGGFace
 from deepface.commons import functions
@@ -13,7 +12,7 @@
 # pylint: disable=line-too-long
 # --------------------------
 # dependency configurations
-tf_version = int(tf.__version__.split(".", maxsplit=1)[0])
+tf_version = functions.get_tf_major_version()
 
 if tf_version == 1:
     from keras.models import Model, Sequential

deepface/models/Demography.py

@@ -1,10 +1,9 @@
 from typing import Union
 from abc import ABC, abstractmethod
 import numpy as np
-import tensorflow as tf
-
-tf_version = int(tf.__version__.split(".", maxsplit=1)[0])
+from deepface.commons import functions
 
+tf_version = functions.get_tf_major_version()
 if tf_version == 1:
     from keras.models import Model
 else:

deepface/models/FacialRecognition.py

@@ -1,9 +1,9 @@
 from abc import ABC
 from typing import Any, Union
 import numpy as np
-import tensorflow as tf
+from deepface.commons import functions
 
-tf_version = int(tf.__version__.split(".", maxsplit=1)[0])
+tf_version = functions.get_tf_major_version()
 if tf_version == 2:
     from tensorflow.keras.models import Model
 else: