Merge pull request serengil#857 from Villyli/master

change model.predict() to model(), notably improve memory leak (issue serengil#697)
Neil-Huang · Oct 22, 2023 · 68eaafc · 68eaafc
2 parents cc5b18e + 1d57b2f
commit 68eaafc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/deepface/DeepFace.py b/deepface/DeepFace.py
@@ -696,7 +696,7 @@ def represent(
  # represent
  if "keras" in str(type(model)):
  # new tf versions show progress bar and it is annoying
- embedding = model.predict(img, verbose=0)[0].tolist()
+ embedding = model(img, training=False).numpy()[0].tolist()
  else:
  # SFace and Dlib are not keras models and no verbose arguments
  embedding = model.predict(img)[0].tolist()