From 51d10000c7d14ca0ecd3df255a31bc91503d7500 Mon Sep 17 00:00:00 2001 From: Sefik Ilkin Serengil Date: Sun, 23 Oct 2022 17:55:10 +0100 Subject: [PATCH] verboss off for just tf models --- deepface/DeepFace.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/deepface/DeepFace.py b/deepface/DeepFace.py index 81f1c22..9394857 100644 --- a/deepface/DeepFace.py +++ b/deepface/DeepFace.py @@ -765,7 +765,11 @@ def represent(img_path, model_name = 'VGG-Face', model = None, enforce_detection #--------------------------------- #represent - embedding = model.predict(img, verbose=0)[0].tolist() + if "keras" in str(type(model)): + embedding = model.predict(img, verbose=0)[0].tolist() + else: + #SFace is not a keras model and it has no verbose argument + embedding = model.predict(img)[0].tolist() return embedding