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

Merge pull request #1151 from serengil/feat-task-2803-facenet-documentation

Browse Source 
docstrings updated
This commit is contained in:
Sefik Ilkin Serengil 2024-03-28 18:50:36 +00:00 committed by GitHub
commit 0f89272532
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 9 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
deepface/basemodels/Facenet.py
View File

@ -97,9 +97,14 @@ def scaling(x, scale):
return x * scale return x * scale
def InceptionResNetV2(dimension: int = 128) -> Model: def InceptionResNetV1(dimension: int = 128) -> Model:
""" """
InceptionResNetV2 model InceptionResNetV1 model heavily inspired from
github.com/davidsandberg/facenet/blob/master/src/models/inception_resnet_v1.py
As mentioned in Sandberg's repo's readme, pre-trained models are using Inception ResNet v1
Besides training process is documented at
sefiks.com/2018/09/03/face-recognition-with-facenet-in-keras/
Args: Args:
dimension (int): number of dimensions in the embedding layer dimension (int): number of dimensions in the embedding layer
Returns: Returns:
@ -1685,7 +1690,7 @@ def load_facenet128d_model(
Returns: Returns:
model (Model) model (Model)
""" """
model = InceptionResNetV2() model = InceptionResNetV1()
# ----------------------------------- # -----------------------------------
@ -1715,7 +1720,7 @@ def load_facenet512d_model(
model (Model) model (Model)
""" """
model = InceptionResNetV2(dimension=512) model = InceptionResNetV1(dimension=512)
# ------------------------- # -------------------------