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

Update DeepFace.py

Browse Source 
model type check
This commit is contained in:
Sefik Ilkin Serengil 2023-12-18 12:06:49 +00:00 committed by GitHub
parent 19f7cf620d
commit 0c0dfb8090
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
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

@ -743,7 +743,8 @@ def represent(
img = functions.normalize_input(img=img, normalization=normalization)
# represent
if "keras" in str(type(model)):
# if "keras" in str(type(model)):
if isinstance(model, Model):
# 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()