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

docstrings updated

Browse Source
This commit is contained in:
Sefik Ilkin Serengil 2024-03-28 18:39:26 +00:00
parent 329606ffe8
commit c22152f921
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
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:
dimension (int): number of dimensions in the embedding layer
Returns:
@ -1685,7 +1690,7 @@ def load_facenet128d_model(
Returns:
model (Model)
"""
model = InceptionResNetV2()
model = InceptionResNetV1()
# -----------------------------------
@ -1715,7 +1720,7 @@ def load_facenet512d_model(
model (Model)
"""
model = InceptionResNetV2(dimension=512)
model = InceptionResNetV1(dimension=512)
# -------------------------