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

fixed for issue 470

Browse Source
This commit is contained in:
Sefik Ilkin Serengil 2022-05-07 20:34:27 +01:00 committed by GitHub
parent fb68d4a8f8
commit e28d900a88
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
tests/face-recognition-how.py
View File

@ -13,7 +13,10 @@ model = VGGFace.loadModel()
#model = OpenFace.loadModel() #model = OpenFace.loadModel()
#model = FbDeepFace.loadModel() #model = FbDeepFace.loadModel()
input_shape = model.layers[0].input_shape[1:3] try:
input_shape = model.layers[0].input_shape[1:3]
except: #issue 470
input_shape = model.layers[0].input_shape[0][1:3]
print("model input shape: ", model.layers[0].input_shape[1:]) print("model input shape: ", model.layers[0].input_shape[1:])
print("model output shape: ", model.layers[-1].input_shape[-1]) print("model output shape: ", model.layers[-1].input_shape[-1])