github/deepface
1
1
Fork 0
mirror of https://github.com/serengil/deepface.git synced 2025-06-09 12:57:08 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix basemodel for vggface

Browse Source
This commit is contained in:
NISH1001 2020-05-20 19:13:13 +05:45
parent 2c0a507994
commit d9326b95bc
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
deepface/basemodels/VGGFace.py
View File

@ -16,7 +16,7 @@ import gdown
# --------------------------------------- # ---------------------------------------
def get_base_model(): def baseModel():
model = Sequential() model = Sequential()
model.add(ZeroPadding2D((1, 1), input_shape=(224, 224, 3))) model.add(ZeroPadding2D((1, 1), input_shape=(224, 224, 3)))
model.add(Convolution2D(64, (3, 3), activation="relu")) model.add(Convolution2D(64, (3, 3), activation="relu"))
@ -86,7 +86,7 @@ def loadModel(model_path=""):
# ----------------------------------- # -----------------------------------
print(f"Loading model from {model_path}") print(f"Loading model from {model_path}")
model = get_base_model() model = baseModel()
model.load_weights(model_path) model.load_weights(model_path)
# ----------------------------------- # -----------------------------------