diff --git a/bob/bio/face/config/baseline/arcface_insightface.py b/bob/bio/face/config/baseline/arcface_insightface.py index b65ea11ed686aa57909fd10ad7aa9aa99515d76f..311672c6c7ecba11a2d1d4d9f531b42dcf8a8a11 100644 --- a/bob/bio/face/config/baseline/arcface_insightface.py +++ b/bob/bio/face/config/baseline/arcface_insightface.py @@ -1,4 +1,4 @@ -from bob.bio.face.embeddings.mxnet import arcface_baseline +from bob.bio.face.embeddings.mxnet import arcface_insightFace_lresnet100 from bob.bio.face.utils import lookup_config_from_database annotation_type, fixed_positions, memory_demanding = lookup_config_from_database( @@ -8,10 +8,10 @@ annotation_type, fixed_positions, memory_demanding = lookup_config_from_database def load(annotation_type, fixed_positions=None): - return arcface_baseline( - embedding=ArcFaceInsightFace_LResNet100(memory_demanding=memory_demanding), + return arcface_insightFace_lresnet100( annotation_type=annotation_type, fixed_positions=fixed_positions, + memory_demanding=memory_demanding, ) diff --git a/bob/bio/face/embeddings/mxnet.py b/bob/bio/face/embeddings/mxnet.py index 5d14bed7592c4c3bff9d075452e3e6deae5643f5..8bb1bde60f9634cfd2603fe8191704ca15893faa 100644 --- a/bob/bio/face/embeddings/mxnet.py +++ b/bob/bio/face/embeddings/mxnet.py @@ -159,7 +159,7 @@ from bob.bio.base.pipelines.vanilla_biometrics import ( ) -def arcface_baseline(embedding, annotation_type, fixed_positions=None): +def arcface_template(embedding, annotation_type, fixed_positions=None): # DEFINE CROPPING cropped_image_size = (112, 112) if annotation_type == "eyes-center": @@ -184,3 +184,12 @@ def arcface_baseline(embedding, annotation_type, fixed_positions=None): return VanillaBiometricsPipeline(transformer, algorithm) + +def arcface_insightFace_lresnet100( + annotation_type, fixed_positions=None, memory_demanding=False +): + return arcface_template( + ArcFaceInsightFace_LResNet100(memory_demanding=memory_demanding), + annotation_type, + fixed_positions, + )