From edcea8f19f07c1d95725915a76e43ccab34a9bcf Mon Sep 17 00:00:00 2001
From: Tiago Freitas Pereira <tiagofrepereira@gmail.com>
Date: Wed, 2 Jun 2021 15:51:25 +0200
Subject: [PATCH] Update ARCFace template
---
bob/bio/face/config/baseline/arcface_insightface.py | 6 +++---
bob/bio/face/embeddings/mxnet.py | 11 ++++++++++-
2 files changed, 13 insertions(+), 4 deletions(-)
diff --git a/bob/bio/face/config/baseline/arcface_insightface.py b/bob/bio/face/config/baseline/arcface_insightface.py
index b65ea11e..311672c6 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 5d14bed7..8bb1bde6 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,
+ )
--
GitLab