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

fix: check first the type of the input_shape before get the shape. Also fix the if-else model_name

Browse Source
This commit is contained in:
danielgo13 2020-06-18 18:30:13 +07:00
parent 0b7873ad55
commit e09d277be9
1 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
deepface/DeepFace.py
View File

@ -196,7 +196,7 @@ def verify(img1_path, img2_path=''
print("Using FB DeepFace model backend", distance_metric,"distance.")
model = FbDeepFace.loadModel()
if model_name == 'DeepID':
elif model_name == 'DeepID':
print("Using DeepID2 model backend", distance_metric,"distance.")
model = DeepID.loadModel()
@ -207,8 +207,11 @@ def verify(img1_path, img2_path=''
#------------------------------
#face recognition models have different size of inputs
input_shape = model.layers[0].input_shape[1:3]
input_shape = model.layers[0].input_shape
if type(input_shape) is list:
input_shape = input_shape[0][1:3]
input_shape_x = input_shape[0]
input_shape_y = input_shape[1]