Skip to content
Snippets Groups Projects
Commit edcea8f1 authored by Tiago de Freitas Pereira's avatar Tiago de Freitas Pereira
Browse files

Update ARCFace template

parent 172e1259
No related branches found
No related tags found
1 merge request!112Feature extractors
Pipeline #51196 failed
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 from bob.bio.face.utils import lookup_config_from_database
annotation_type, fixed_positions, memory_demanding = 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 ...@@ -8,10 +8,10 @@ annotation_type, fixed_positions, memory_demanding = lookup_config_from_database
def load(annotation_type, fixed_positions=None): def load(annotation_type, fixed_positions=None):
return arcface_baseline( return arcface_insightFace_lresnet100(
embedding=ArcFaceInsightFace_LResNet100(memory_demanding=memory_demanding),
annotation_type=annotation_type, annotation_type=annotation_type,
fixed_positions=fixed_positions, fixed_positions=fixed_positions,
memory_demanding=memory_demanding,
) )
......
...@@ -159,7 +159,7 @@ from bob.bio.base.pipelines.vanilla_biometrics import ( ...@@ -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 # DEFINE CROPPING
cropped_image_size = (112, 112) cropped_image_size = (112, 112)
if annotation_type == "eyes-center": if annotation_type == "eyes-center":
...@@ -184,3 +184,12 @@ def arcface_baseline(embedding, annotation_type, fixed_positions=None): ...@@ -184,3 +184,12 @@ def arcface_baseline(embedding, annotation_type, fixed_positions=None):
return VanillaBiometricsPipeline(transformer, algorithm) 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,
)
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment