Skip to content
Snippets Groups Projects
Commit 9100d0db authored by Laurent COLBOIS's avatar Laurent COLBOIS
Browse files

Refactor reading from the database configuration

parent c2a907cc
Branches
Tags
1 merge request!119Refactor baseline config helpers
Pipeline #50565 failed
Showing
with 77 additions and 152 deletions
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):
......
......@@ -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(),
)
......
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):
......
......@@ -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):
......
......@@ -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,
......
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):
......
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):
......
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):
......
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):
......
......@@ -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 ##########
......
......@@ -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):
......
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):
......
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):
......
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):
......
......@@ -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):
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment