diff --git a/bob/bio/face/config/baseline/arcface_insightface.py b/bob/bio/face/config/baseline/arcface_insightface.py index 3aae89491d77909db49a10d3c8ef010d66e37f4a..b2de000f4d30dbbbdc7c34592f19b249e69b976c 100644 --- a/bob/bio/face/config/baseline/arcface_insightface.py +++ b/bob/bio/face/config/baseline/arcface_insightface.py @@ -1,21 +1,14 @@ 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): diff --git a/bob/bio/face/config/baseline/dummy.py b/bob/bio/face/config/baseline/dummy.py index 7e0317a3f6a452a83efb5d2188f48081e71c386b..eaa993d3917315b1e1bfb4e83290c07e2d858202 100644 --- a/bob/bio/face/config/baseline/dummy.py +++ b/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(), ) diff --git a/bob/bio/face/config/baseline/facenet_sanderberg.py b/bob/bio/face/config/baseline/facenet_sanderberg.py index e3d1dc269922f9633feceab0b988514ff42b37d2..20e3651aab7a2701e5b670481beb46e6126e4b79 100644 --- a/bob/bio/face/config/baseline/facenet_sanderberg.py +++ b/bob/bio/face/config/baseline/facenet_sanderberg.py @@ -1,23 +1,16 @@ 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): diff --git a/bob/bio/face/config/baseline/gabor_graph.py b/bob/bio/face/config/baseline/gabor_graph.py index 1f1a786060b6041c2549ac55436e6e623be495bd..6c79bffd407e26b327ca22574ba9f5c08b84b7a8 100644 --- a/bob/bio/face/config/baseline/gabor_graph.py +++ b/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): diff --git a/bob/bio/face/config/baseline/helpers.py b/bob/bio/face/config/baseline/helpers.py index aa9e7cd481aba706af4ce8c22b7cc10c5d343128..8c04bcc2a7fe30f5e67b0f4086b36225e7331daf 100644 --- a/bob/bio/face/config/baseline/helpers.py +++ b/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, diff --git a/bob/bio/face/config/baseline/inception_resnetv1_casiawebface.py b/bob/bio/face/config/baseline/inception_resnetv1_casiawebface.py index acc6f981aecc0ae7312097e68536055605c50ce6..90bd0111248cfd6ebd07bb7f9eb7642b44f41c11 100644 --- a/bob/bio/face/config/baseline/inception_resnetv1_casiawebface.py +++ b/bob/bio/face/config/baseline/inception_resnetv1_casiawebface.py @@ -1,23 +1,16 @@ 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): diff --git a/bob/bio/face/config/baseline/inception_resnetv1_msceleb.py b/bob/bio/face/config/baseline/inception_resnetv1_msceleb.py index 70a1a58fdee70f1566a1bfb63e10493cf5599809..c459709626a019701667d3276ad06e1f8873d6a8 100644 --- a/bob/bio/face/config/baseline/inception_resnetv1_msceleb.py +++ b/bob/bio/face/config/baseline/inception_resnetv1_msceleb.py @@ -1,23 +1,16 @@ 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): diff --git a/bob/bio/face/config/baseline/inception_resnetv2_casiawebface.py b/bob/bio/face/config/baseline/inception_resnetv2_casiawebface.py index 3dd2704330cadf49873af96489cc2d9d51e27cfd..cb82dee8f8e1f4a516721fd33f06a984a30b65f5 100644 --- a/bob/bio/face/config/baseline/inception_resnetv2_casiawebface.py +++ b/bob/bio/face/config/baseline/inception_resnetv2_casiawebface.py @@ -1,23 +1,16 @@ 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): diff --git a/bob/bio/face/config/baseline/inception_resnetv2_msceleb.py b/bob/bio/face/config/baseline/inception_resnetv2_msceleb.py index ba339bcf11dd9928ed37f1f9d95eeba7f559f8fc..fa502308c7629ba3d2d8487582cce6428603c4eb 100644 --- a/bob/bio/face/config/baseline/inception_resnetv2_msceleb.py +++ b/bob/bio/face/config/baseline/inception_resnetv2_msceleb.py @@ -1,23 +1,16 @@ 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): diff --git a/bob/bio/face/config/baseline/lda.py b/bob/bio/face/config/baseline/lda.py index 1343f51a431ee7b97d0ca704c9ec910f57c1230d..062f18f6423734e9e4d7fa4dcbc881a8ec4a093a 100644 --- a/bob/bio/face/config/baseline/lda.py +++ b/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 ########## diff --git a/bob/bio/face/config/baseline/lgbphs.py b/bob/bio/face/config/baseline/lgbphs.py index eebcbba56258d6a85e792c4fb4e860be748e0269..902f43cfd55d3de77ce14f426bce118d1e5ae98d 100644 --- a/bob/bio/face/config/baseline/lgbphs.py +++ b/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): diff --git a/bob/bio/face/config/baseline/mobilenetv2_msceleb_arcface_2021.py b/bob/bio/face/config/baseline/mobilenetv2_msceleb_arcface_2021.py index e68e7ef3edf5d8c080efc9aac7bf06a39d85982b..aae95a2a7d5aed639dba7c308b1d54cbb89e99b8 100644 --- a/bob/bio/face/config/baseline/mobilenetv2_msceleb_arcface_2021.py +++ b/bob/bio/face/config/baseline/mobilenetv2_msceleb_arcface_2021.py @@ -1,21 +1,14 @@ 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): diff --git a/bob/bio/face/config/baseline/resnet50_msceleb_arcface_2021.py b/bob/bio/face/config/baseline/resnet50_msceleb_arcface_2021.py index dfeb4b74af2958879e83c74b4428c2cb7601d308..f60903ace50188014553d8a6d25ebc06ef0cdbd4 100644 --- a/bob/bio/face/config/baseline/resnet50_msceleb_arcface_2021.py +++ b/bob/bio/face/config/baseline/resnet50_msceleb_arcface_2021.py @@ -1,21 +1,14 @@ 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): diff --git a/bob/bio/face/config/baseline/resnet50_vgg2_arcface_2021.py b/bob/bio/face/config/baseline/resnet50_vgg2_arcface_2021.py index b8f13ec4834df8198225b66f3e1dfa0f22ed7f7b..0a6941d0e9c670a65dc7605dac0c5f38bc5b1788 100644 --- a/bob/bio/face/config/baseline/resnet50_vgg2_arcface_2021.py +++ b/bob/bio/face/config/baseline/resnet50_vgg2_arcface_2021.py @@ -1,21 +1,14 @@ 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): diff --git a/bob/bio/face/config/baseline/tf2_inception_resnet.py b/bob/bio/face/config/baseline/tf2_inception_resnet.py index 9804831574e62e555bc70213915f340f965bba17..f71e0a78bdd43a141028687b0e7acd70ee70cabb 100644 --- a/bob/bio/face/config/baseline/tf2_inception_resnet.py +++ b/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):