From 1d57b2f7f3d71ab013ccf0b11d71d6cd4a6c8e74 Mon Sep 17 00:00:00 2001 From: lilw5 Date: Tue, 10 Oct 2023 11:16:44 +0800 Subject: [PATCH] change model.predict() to model(), improve mem leak notably --- deepface/DeepFace.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/deepface/DeepFace.py b/deepface/DeepFace.py index 35e179b..15d6663 100644 --- 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()