change model.predict() to model(), improve mem leak notably

Neil-Huang · Oct 10, 2023 · 1d57b2f · 1d57b2f
1 parent fb8924e
commit 1d57b2f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/deepface/DeepFace.py b/deepface/DeepFace.py
@@ -693,7 +693,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()