From 2895148ff2c251153d087afa8cf7c886ca1645d8 Mon Sep 17 00:00:00 2001
From: Tiago Freitas Pereira <tiagofrepereira@gmail.com>
Date: Wed, 20 May 2020 14:11:42 +0200
Subject: [PATCH] This might have solved the dask issue
---
bob/ip/tensorflow_extractor/FaceNet.py | 11 ++++++++---
1 file changed, 8 insertions(+), 3 deletions(-)
diff --git a/bob/ip/tensorflow_extractor/FaceNet.py b/bob/ip/tensorflow_extractor/FaceNet.py
index 9895ebd..e7bfb3d 100644
--- a/bob/ip/tensorflow_extractor/FaceNet.py
+++ b/bob/ip/tensorflow_extractor/FaceNet.py
@@ -107,8 +107,14 @@ class FaceNet(object):
return img[None, ...]
def load_model(self):
+ tf.compat.v1.reset_default_graph()
+
+ session_conf = tf.ConfigProto(
+ intra_op_parallelism_threads=1,
+ inter_op_parallelism_threads=1)
+
self.graph = tf.Graph()
- self.session = tf.compat.v1.Session(graph=self.graph)
+ self.session = tf.compat.v1.Session(graph=self.graph, config=session_conf)
if self.model_path is None:
self.model_path = self.get_modelpath()
@@ -188,8 +194,7 @@ class FaceNet(object):
def __setstate__(self, d):
# Handling unpicklable objects
- self.__dict__ = d
- self.load_model()
+ self.__dict__ = d
def __getstate__(self):
# Handling unpicklable objects
--
GitLab