From d9326b95bcaeaf381f2b552b06e118890ab95255 Mon Sep 17 00:00:00 2001 From: NISH1001 Date: Wed, 20 May 2020 19:13:13 +0545 Subject: [PATCH] Fix basemodel for vggface --- deepface/basemodels/VGGFace.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/deepface/basemodels/VGGFace.py b/deepface/basemodels/VGGFace.py index 5b66b5a..f8395bb 100644 --- a/deepface/basemodels/VGGFace.py +++ b/deepface/basemodels/VGGFace.py @@ -16,7 +16,7 @@ import gdown # --------------------------------------- -def get_base_model(): +def baseModel(): model = Sequential() model.add(ZeroPadding2D((1, 1), input_shape=(224, 224, 3))) model.add(Convolution2D(64, (3, 3), activation="relu")) @@ -86,7 +86,7 @@ def loadModel(model_path=""): # ----------------------------------- print(f"Loading model from {model_path}") - model = get_base_model() + model = baseModel() model.load_weights(model_path) # -----------------------------------