Commit c22152f 1 parent 329606f commit c22152f Copy full SHA for c22152f
File tree 1 file changed +9
-4
lines changed
1 file changed +9
-4
lines changed Original file line number Diff line number Diff line change @@ -97,9 +97,14 @@ def scaling(x, scale):
97
97
return x * scale
98
98
99
99
100
- def InceptionResNetV2 (dimension : int = 128 ) -> Model :
100
+ def InceptionResNetV1 (dimension : int = 128 ) -> Model :
101
101
"""
102
- InceptionResNetV2 model
102
+ InceptionResNetV1 model heavily inspired from
103
+ github.com/davidsandberg/facenet/blob/master/src/models/inception_resnet_v1.py
104
+ As mentioned in Sandberg's repo's readme, pre-trained models are using Inception ResNet v1
105
+ Besides training process is documented at
106
+ sefiks.com/2018/09/03/face-recognition-with-facenet-in-keras/
107
+
103
108
Args:
104
109
dimension (int): number of dimensions in the embedding layer
105
110
Returns:
@@ -1685,7 +1690,7 @@ def load_facenet128d_model(
1685
1690
Returns:
1686
1691
model (Model)
1687
1692
"""
1688
- model = InceptionResNetV2 ()
1693
+ model = InceptionResNetV1 ()
1689
1694
1690
1695
# -----------------------------------
1691
1696
@@ -1715,7 +1720,7 @@ def load_facenet512d_model(
1715
1720
model (Model)
1716
1721
"""
1717
1722
1718
- model = InceptionResNetV2 (dimension = 512 )
1723
+ model = InceptionResNetV1 (dimension = 512 )
1719
1724
1720
1725
# -------------------------
1721
1726
You can’t perform that action at this time.
0 commit comments