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

Merge pull request #857 from Villyli/master

Browse Source 
change model.predict() to model(), notably improve memory leak (issue #697)
This commit is contained in:
Sefik Ilkin Serengil 2023-10-22 19:41:11 +01:00 committed by GitHub
commit 68eaafc614
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
deepface/DeepFace.py
View File

@ -696,7 +696,7 @@ 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 # 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: else:
# SFace and Dlib are not keras models and no verbose arguments # SFace and Dlib are not keras models and no verbose arguments
embedding = model.predict(img)[0].tolist() embedding = model.predict(img)[0].tolist()