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
Branches
Tags
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
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,
)
......
......@@ -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,
)
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment