github/deepface
1
1
Fork 0
mirror of https://github.com/serengil/deepface.git synced 2025-06-07 20:15:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

verboss off for just tf models

Browse Source
This commit is contained in:
Sefik Ilkin Serengil 2022-10-23 17:55:10 +01:00
parent bbc3c6d59b
commit 51d10000c7
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
deepface/DeepFace.py
View File

@ -765,7 +765,11 @@ def represent(img_path, model_name = 'VGG-Face', model = None, enforce_detection
#--------------------------------- #---------------------------------
#represent #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 return embedding