Merge pull request #1456 from timselier/patch-1

Update VGGFace.py
This commit is contained in:
Sefik Ilkin Serengil 2025-04-01 11:07:31 +01:00 committed by GitHub
commit 96a7b98f33
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -161,6 +161,6 @@ def load_model(
# base_model_output = Lambda(lambda x: K.l2_normalize(x, axis=1), name="norm_layer")(
# base_model_output
# )
vgg_face_descriptor = Model(inputs=model.input, outputs=base_model_output)
vgg_face_descriptor = Model(inputs=model.layers[0].input, outputs=base_model_output)
return vgg_face_descriptor