diff --git a/deepface/DeepFace.py b/deepface/DeepFace.py index 067a1b5..259bf19 100644 --- a/deepface/DeepFace.py +++ b/deepface/DeepFace.py @@ -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]