Refactor reading from the database configuration

bob/bio/face/config/baseline/arcface_insightface.py

 from bob.bio.face.embeddings.mxnet_models import ArcFaceInsightFace
-from bob.bio.face.config.baseline.helpers import embedding_transformer_112x112
+from bob.bio.face.config.baseline.helpers import (
+    embedding_transformer_112x112,
+    lookup_config_from_database,
+)
 from bob.bio.base.pipelines.vanilla_biometrics import (
     Distance,
     VanillaBiometricsPipeline,
 )
 
-
-if "database" in locals():
-    annotation_type = database.annotation_type
-    fixed_positions = database.fixed_positions
-    memory_demanding = (
-        database.memory_demanding if hasattr(database, "memory_demanding") else False
-    )
-else:
-    annotation_type = None
-    fixed_positions = None
-    memory_demanding = False
+annotation_type, fixed_positions, memory_demanding = lookup_config_from_database()
 
 
 def load(annotation_type, fixed_positions=None):

bob/bio/face/config/baseline/dummy.py

@@ -5,21 +5,17 @@ from bob.bio.base.pipelines.vanilla_biometrics import (
     VanillaBiometricsPipeline,
 )
 from bob.pipelines.transformers import SampleLinearize
+from bob.bio.face.config.baseline.helpers import lookup_config_from_database
 
-if "database" in locals():
-    annotation_type = database.annotation_type
-    fixed_positions = database.fixed_positions
-else:
-    annotation_type = None
-    fixed_positions = None
-
+annotation_type, fixed_positions, memory_demanding = lookup_config_from_database()
 
 import bob.ip.color
 from sklearn.base import TransformerMixin, BaseEstimator
-class ToGray(TransformerMixin, BaseEstimator):
 
+
+class ToGray(TransformerMixin, BaseEstimator):
     def transform(self, X, annotations=None):
-        return [bob.ip.color.rgb_to_gray(data)[0:10,0:10] for data in X]
+        return [bob.ip.color.rgb_to_gray(data)[0:10, 0:10] for data in X]
 
     def _more_tags(self):
         return {"stateless": True, "requires_fit": False}
@@ -34,9 +30,7 @@ def load(annotation_type, fixed_positions=None):
 
     transformer = make_pipeline(
         wrap(
-            ["sample"],
-            ToGray(),
-            transform_extra_arguments=transform_extra_arguments,
+            ["sample"], ToGray(), transform_extra_arguments=transform_extra_arguments,
         ),
         SampleLinearize(),
     )

bob/bio/face/config/baseline/facenet_sanderberg.py

 from bob.bio.face.embeddings.tf2_inception_resnet import (
     FaceNetSanderberg_20170512_110547,
 )
-from bob.bio.face.config.baseline.helpers import embedding_transformer_160x160
+from bob.bio.face.config.baseline.helpers import (
+    embedding_transformer_160x160,
+    lookup_config_from_database,
+)
 from bob.bio.base.pipelines.vanilla_biometrics import (
     Distance,
     VanillaBiometricsPipeline,
 )
 
-memory_demanding = False
-if "database" in locals():
-    annotation_type = database.annotation_type
-    fixed_positions = database.fixed_positions
-    memory_demanding = (
-        database.memory_demanding if hasattr(database, "memory_demanding") else False
-    )
-
-else:
-    annotation_type = None
-    fixed_positions = None
+annotation_type, fixed_positions, memory_demanding = lookup_config_from_database()
 
 
 def load(annotation_type, fixed_positions=None):

bob/bio/face/config/baseline/gabor_graph.py

@@ -3,7 +3,7 @@ from bob.bio.base.pipelines.vanilla_biometrics import (
     VanillaBiometricsPipeline,
     BioAlgorithmLegacy,
 )
-from bob.bio.face.config.baseline.helpers import crop_80x64
+from bob.bio.face.config.baseline.helpers import crop_80x64, lookup_config_from_database
 import math
 import numpy as np
 import bob.bio.face
@@ -17,12 +17,7 @@ import logging
 logger = logging.getLogger(__name__)
 
 #### SOLVING IF THERE'S ANY DATABASE INFORMATION
-if "database" in locals():
-    annotation_type = database.annotation_type
-    fixed_positions = database.fixed_positions
-else:
-    annotation_type = None
-    fixed_positions = None
+annotation_type, fixed_positions, memory_demanding = lookup_config_from_database()
 
 
 def get_cropper(annotation_type, fixed_positions=None):

bob/bio/face/config/baseline/helpers.py

@@ -9,6 +9,24 @@ import logging
 logger = logging.getLogger(__name__)
 
 
+def lookup_config_from_database():
+    if "database" in locals():
+        annotation_type = database.annotation_type
+        fixed_positions = database.fixed_positions
+        memory_demanding = (
+            database.memory_demanding
+            if hasattr(database, "memory_demanding")
+            else False
+        )
+
+    else:
+        annotation_type = None
+        fixed_positions = None
+        memory_demanding = False
+
+    return annotation_type, fixed_positions, memory_demanding
+
+
 def face_crop_solver(
     cropped_image_size,
     cropped_positions=None,

bob/bio/face/config/baseline/inception_resnetv1_casiawebface.py

 from bob.bio.face.embeddings.tf2_inception_resnet import (
     InceptionResnetv1_Casia_CenterLoss_2018,
 )
-from bob.bio.face.config.baseline.helpers import embedding_transformer_160x160
+from bob.bio.face.config.baseline.helpers import (
+    embedding_transformer_160x160,
+    lookup_config_from_database,
+)
 from bob.bio.base.pipelines.vanilla_biometrics import (
     Distance,
     VanillaBiometricsPipeline,
 )
 
-memory_demanding = False
-if "database" in locals():
-    annotation_type = database.annotation_type
-    fixed_positions = database.fixed_positions
-    memory_demanding = (
-        database.memory_demanding if hasattr(database, "memory_demanding") else False
-    )
-
-else:
-    annotation_type = None
-    fixed_positions = None
+annotation_type, fixed_positions, memory_demanding = lookup_config_from_database()
 
 
 def load(annotation_type, fixed_positions=None):

bob/bio/face/config/baseline/inception_resnetv1_msceleb.py

 from bob.bio.face.embeddings.tf2_inception_resnet import (
     InceptionResnetv1_MsCeleb_CenterLoss_2018,
 )
-from bob.bio.face.config.baseline.helpers import embedding_transformer_160x160
+from bob.bio.face.config.baseline.helpers import (
+    embedding_transformer_160x160,
+    lookup_config_from_database,
+)
 from bob.bio.base.pipelines.vanilla_biometrics import (
     Distance,
     VanillaBiometricsPipeline,
 )
 
-memory_demanding = False
-if "database" in locals():
-    annotation_type = database.annotation_type
-    fixed_positions = database.fixed_positions
-    memory_demanding = (
-        database.memory_demanding if hasattr(database, "memory_demanding") else False
-    )
-
-else:
-    annotation_type = None
-    fixed_positions = None
+annotation_type, fixed_positions, memory_demanding = lookup_config_from_database()
 
 
 def load(annotation_type, fixed_positions=None):

bob/bio/face/config/baseline/inception_resnetv2_casiawebface.py

 from bob.bio.face.embeddings.tf2_inception_resnet import (
     InceptionResnetv2_Casia_CenterLoss_2018,
 )
-from bob.bio.face.config.baseline.helpers import embedding_transformer_160x160
+from bob.bio.face.config.baseline.helpers import (
+    embedding_transformer_160x160,
+    lookup_config_from_database,
+)
 from bob.bio.base.pipelines.vanilla_biometrics import (
     Distance,
     VanillaBiometricsPipeline,
 )
 
-memory_demanding = False
-if "database" in locals():
-    annotation_type = database.annotation_type
-    fixed_positions = database.fixed_positions
-    memory_demanding = (
-        database.memory_demanding if hasattr(database, "memory_demanding") else False
-    )
-
-else:
-    annotation_type = None
-    fixed_positions = None
+annotation_type, fixed_positions, memory_demanding = lookup_config_from_database()
 
 
 def load(annotation_type, fixed_positions=None):

bob/bio/face/config/baseline/inception_resnetv2_msceleb.py

 from bob.bio.face.embeddings.tf2_inception_resnet import (
     InceptionResnetv2_MsCeleb_CenterLoss_2018,
 )
-from bob.bio.face.config.baseline.helpers import embedding_transformer_160x160
+from bob.bio.face.config.baseline.helpers import (
+    embedding_transformer_160x160,
+    lookup_config_from_database,
+)
 from bob.bio.base.pipelines.vanilla_biometrics import (
     Distance,
     VanillaBiometricsPipeline,
 )
 
-memory_demanding = False
-if "database" in locals():
-    annotation_type = database.annotation_type
-    fixed_positions = database.fixed_positions
-
-    memory_demanding = (
-        database.memory_demanding if hasattr(database, "memory_demanding") else False
-    )
-else:
-    annotation_type = None
-    fixed_positions = None
+annotation_type, fixed_positions, memory_demanding = lookup_config_from_database()
 
 
 def load(annotation_type, fixed_positions=None):

bob/bio/face/config/baseline/lda.py

@@ -3,7 +3,7 @@ from bob.bio.base.pipelines.vanilla_biometrics import (
     VanillaBiometricsPipeline,
     BioAlgorithmLegacy,
 )
-from bob.bio.face.config.baseline.helpers import crop_80x64
+from bob.bio.face.config.baseline.helpers import crop_80x64, lookup_config_from_database
 import numpy as np
 import bob.bio.face
 from sklearn.pipeline import make_pipeline
@@ -18,12 +18,7 @@ import logging
 logger = logging.getLogger(__name__)
 
 #### SOLVING IF THERE'S ANY DATABASE INFORMATION
-if "database" in locals():
-    annotation_type = database.annotation_type
-    fixed_positions = database.fixed_positions
-else:
-    annotation_type = None
-    fixed_positions = None
+annotation_type, fixed_positions, memory_demanding = lookup_config_from_database()
 
 
 ####### SOLVING THE FACE CROPPER TO BE USED ##########

bob/bio/face/config/baseline/lgbphs.py

@@ -3,7 +3,7 @@ from bob.bio.base.pipelines.vanilla_biometrics import (
     VanillaBiometricsPipeline,
     BioAlgorithmLegacy,
 )
-from bob.bio.face.config.baseline.helpers import crop_80x64
+from bob.bio.face.config.baseline.helpers import crop_80x64, lookup_config_from_database
 import math
 import numpy as np
 import bob.bio.face
@@ -13,12 +13,7 @@ import bob.math
 
 
 #### SOLVING IF THERE'S ANY DATABASE INFORMATION
-if "database" in locals():
-    annotation_type = database.annotation_type
-    fixed_positions = database.fixed_positions
-else:
-    annotation_type = None
-    fixed_positions = None
+annotation_type, fixed_positions, memory_demanding = lookup_config_from_database()
 
 
 def get_cropper(annotation_type, fixed_positions=None):

bob/bio/face/config/baseline/mobilenetv2_msceleb_arcface_2021.py

 from bob.bio.face.embeddings.mobilenet_v2 import MobileNetv2_MsCeleb_ArcFace_2021
-from bob.bio.face.config.baseline.helpers import embedding_transformer_112x112
+from bob.bio.face.config.baseline.helpers import (
+    embedding_transformer_112x112,
+    lookup_config_from_database,
+)
 from bob.bio.base.pipelines.vanilla_biometrics import (
     Distance,
     VanillaBiometricsPipeline,
 )
 
-memory_demanding = False
-if "database" in locals():
-    annotation_type = database.annotation_type
-    fixed_positions = database.fixed_positions
-
-    memory_demanding = (
-        database.memory_demanding if hasattr(database, "memory_demanding") else False
-    )
-else:
-    annotation_type = None
-    fixed_positions = None
+annotation_type, fixed_positions, memory_demanding = lookup_config_from_database()
 
 
 def load(annotation_type, fixed_positions=None):

bob/bio/face/config/baseline/resnet50_msceleb_arcface_2021.py

 from bob.bio.face.embeddings.resnet50 import Resnet50_MsCeleb_ArcFace_2021
-from bob.bio.face.config.baseline.helpers import embedding_transformer_112x112
+from bob.bio.face.config.baseline.helpers import (
+    embedding_transformer_112x112,
+    lookup_config_from_database,
+)
 from bob.bio.base.pipelines.vanilla_biometrics import (
     Distance,
     VanillaBiometricsPipeline,
 )
 
-memory_demanding = False
-if "database" in locals():
-    annotation_type = database.annotation_type
-    fixed_positions = database.fixed_positions
-
-    memory_demanding = (
-        database.memory_demanding if hasattr(database, "memory_demanding") else False
-    )
-else:
-    annotation_type = None
-    fixed_positions = None
+annotation_type, fixed_positions, memory_demanding = lookup_config_from_database()
 
 
 def load(annotation_type, fixed_positions=None):

bob/bio/face/config/baseline/resnet50_vgg2_arcface_2021.py

 from bob.bio.face.embeddings.resnet50 import Resnet50_VGG2_ArcFace_2021
-from bob.bio.face.config.baseline.helpers import embedding_transformer_112x112
+from bob.bio.face.config.baseline.helpers import (
+    embedding_transformer_112x112,
+    lookup_config_from_database,
+)
 from bob.bio.base.pipelines.vanilla_biometrics import (
     Distance,
     VanillaBiometricsPipeline,
 )
 
-memory_demanding = False
-if "database" in locals():
-    annotation_type = database.annotation_type
-    fixed_positions = database.fixed_positions
-
-    memory_demanding = (
-        database.memory_demanding if hasattr(database, "memory_demanding") else False
-    )
-else:
-    annotation_type = None
-    fixed_positions = None
+annotation_type, fixed_positions, memory_demanding = lookup_config_from_database()
 
 
 def load(annotation_type, fixed_positions=None):

bob/bio/face/config/baseline/tf2_inception_resnet.py

@@ -4,6 +4,7 @@ from bob.bio.face.preprocessor import FaceCrop
 from bob.bio.face.config.baseline.helpers import (
     embedding_transformer_default_cropping,
     embedding_transformer,
+    lookup_config_from_database,
 )
 
 from sklearn.pipeline import make_pipeline
@@ -13,17 +14,7 @@ from bob.bio.base.pipelines.vanilla_biometrics import (
     VanillaBiometricsPipeline,
 )
 
-memory_demanding = False
-if "database" in locals():
-    annotation_type = database.annotation_type
-    fixed_positions = database.fixed_positions
-    memory_demanding = (
-        database.memory_demanding if hasattr(database, "memory_demanding") else False
-    )
-
-else:
-    annotation_type = None
-    fixed_positions = None
+annotation_type, fixed_positions, memory_demanding = lookup_config_from_database()
 
 
 def load(annotation_type, fixed_positions=None):