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

old version of model.predict stored

Browse Source
This commit is contained in:
Sefik Ilkin Serengil 2023-10-22 19:46:02 +01:00
parent 68eaafc614
commit 8eb8d96a54
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
deepface/DeepFace.py
View File

@ -695,7 +695,8 @@ def represent(
# represent # represent
if "keras" in str(type(model)): 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() embedding = model(img, training=False).numpy()[0].tolist()
else: else:
# SFace and Dlib are not keras models and no verbose arguments # SFace and Dlib are not keras models and no verbose arguments