From e28d900a88864bfeaa02144204702f01a5b6dfeb Mon Sep 17 00:00:00 2001 From: Sefik Ilkin Serengil Date: Sat, 7 May 2022 20:34:27 +0100 Subject: [PATCH] fixed for issue 470 --- tests/face-recognition-how.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/tests/face-recognition-how.py b/tests/face-recognition-how.py index 11aa398..00b56b8 100644 --- a/tests/face-recognition-how.py +++ b/tests/face-recognition-how.py @@ -13,7 +13,10 @@ model = VGGFace.loadModel() #model = OpenFace.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 output shape: ", model.layers[-1].input_shape[-1])