From 8ddc6018c573974b31c1468d7a459a7309a226c0 Mon Sep 17 00:00:00 2001
From: Tiago Freitas Pereira <tiagofrepereira@gmail.com>
Date: Mon, 15 Feb 2021 14:31:57 +0100
Subject: [PATCH] Fixed issue with InceptionNetowrks
---
bob/bio/face/embeddings/tf2_inception_resnet.py | 8 ++++++++
1 file changed, 8 insertions(+)
diff --git a/bob/bio/face/embeddings/tf2_inception_resnet.py b/bob/bio/face/embeddings/tf2_inception_resnet.py
index 52ad8d26..b462f3a8 100644
--- a/bob/bio/face/embeddings/tf2_inception_resnet.py
+++ b/bob/bio/face/embeddings/tf2_inception_resnet.py
@@ -80,6 +80,14 @@ class TransformTensorflow(TransformerMixin, BaseEstimator):
d["model"] = None
return d
+ def inference(self, X):
+ if self.preprocessor is not None:
+ X = self.preprocessor(tf.cast(X, "float32"))
+
+ prelogits = self.model.predict_on_batch(X)
+ embeddings = tf.math.l2_normalize(prelogits, axis=-1)
+ return embeddings
+
def _more_tags(self):
return {"stateless": True, "requires_fit": False}
--
GitLab