From 8eb8d96a54a1e8944be9ab12909f7bce767d95f8 Mon Sep 17 00:00:00 2001 From: Sefik Ilkin Serengil Date: Sun, 22 Oct 2023 19:46:02 +0100 Subject: [PATCH] old version of model.predict stored --- deepface/DeepFace.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/deepface/DeepFace.py b/deepface/DeepFace.py index 565c6e5..c61c31b 100644 --- a/deepface/DeepFace.py +++ b/deepface/DeepFace.py @@ -695,7 +695,8 @@ def represent( # represent if "keras" in str(type(model)): - # new tf versions show progress bar and it is annoying + # model.predict causes memory issue when it is called in a for loop + # 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