From 0943c5e77c6f53e38c33826eb8ed87ae313d99aa Mon Sep 17 00:00:00 2001
From: Tiago Freitas Pereira <tiagofrepereira@gmail.com>
Date: Fri, 19 Jun 2020 15:24:22 +0200
Subject: [PATCH] Changed picklable methods for the embeddings
---
bob/bio/face/embeddings/facenet_sanderberg.py | 2 +-
bob/bio/face/embeddings/tensorflow_compat_v1.py | 4 ++--
2 files changed, 3 insertions(+), 3 deletions(-)
diff --git a/bob/bio/face/embeddings/facenet_sanderberg.py b/bob/bio/face/embeddings/facenet_sanderberg.py
index ceaf9a84..ada60509 100644
--- a/bob/bio/face/embeddings/facenet_sanderberg.py
+++ b/bob/bio/face/embeddings/facenet_sanderberg.py
@@ -223,7 +223,6 @@ class FaceNetSanderberg(TransformerMixin, BaseEstimator):
def __setstate__(self, d):
# Handling unpicklable objects
self.__dict__ = d
- self.loaded = False
def __getstate__(self):
import tensorflow as tf
@@ -236,6 +235,7 @@ class FaceNetSanderberg(TransformerMixin, BaseEstimator):
d.pop("images_placeholder") if "images_placeholder" in d else None
d.pop("phase_train_placeholder") if "phase_train_placeholder" in d else None
tf.compat.v1.reset_default_graph()
+ self.loaded = False
return d
def _more_tags(self):
diff --git a/bob/bio/face/embeddings/tensorflow_compat_v1.py b/bob/bio/face/embeddings/tensorflow_compat_v1.py
index 0936523f..5659cb3f 100644
--- a/bob/bio/face/embeddings/tensorflow_compat_v1.py
+++ b/bob/bio/face/embeddings/tensorflow_compat_v1.py
@@ -122,8 +122,7 @@ class TensorflowCompatV1(TransformerMixin, BaseEstimator):
def __setstate__(self, d):
# Handling unpicklable objects
- self.__dict__ = d
- self.loaded = False
+ self.__dict__ = d
def __getstate__(self):
import tensorflow as tf
@@ -134,6 +133,7 @@ class TensorflowCompatV1(TransformerMixin, BaseEstimator):
d.pop("input_tensor", None)
d.pop("embedding", None)
tf.compat.v1.reset_default_graph()
+ self.loaded = False
return d
# def __del__(self):
--
GitLab