Skip to content
Snippets Groups Projects
Commit b2044c7f authored by Tiago de Freitas Pereira's avatar Tiago de Freitas Pereira
Browse files

Attempt to make the __call__/serialization work

parent 3fecd074
No related branches found
No related tags found
1 merge request!17WIP: Added ArcFace model
Pipeline #39970 failed
...@@ -153,7 +153,6 @@ class FaceNet(object): ...@@ -153,7 +153,6 @@ class FaceNet(object):
self.loaded = True self.loaded = True
def __call__(self, img): def __call__(self, img):
# with _semaphore:
images = self._check_feature(img) images = self._check_feature(img)
if not self.loaded: if not self.loaded:
self.load_model() self.load_model()
...@@ -194,8 +193,6 @@ class FaceNet(object): ...@@ -194,8 +193,6 @@ class FaceNet(object):
def __getstate__(self): def __getstate__(self):
# Handling unpicklable objects # Handling unpicklable objects
# with _semaphore:
# self._clean_unpicklables()
self.loaded = False self.loaded = False
d = self.__dict__ d = self.__dict__
d.pop("session") if "session" in d else None d.pop("session") if "session" in d else None
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment