FaceNet to Facenet

serengil · Jan 21, 2024 · c9a7db8 · c9a7db8
1 parent 0d6aa10
commit c9a7db8
Showing 1 changed file with 3 additions and 12 deletions.
diff --git a/deepface/modules/modeling.py b/deepface/modules/modeling.py
@@ -2,16 +2,7 @@
 from typing import Any
 
 # project dependencies
-from deepface.basemodels import (
-    VGGFace,
-    OpenFace,
-    FbDeepFace,
-    DeepID,
-    ArcFace,
-    SFace,
-    Dlib,
-    FaceNet,
-)
+from deepface.basemodels import VGGFace, OpenFace, FbDeepFace, DeepID, ArcFace, SFace, Dlib, Facenet
 from deepface.extendedmodels import Age, Gender, Race, Emotion
 
 
@@ -33,8 +24,8 @@ def build_model(model_name: str) -> Any:
     models = {
         "VGG-Face": VGGFace.VggFaceClient,
         "OpenFace": OpenFace.OpenFaceClient,
-        "Facenet": FaceNet.FaceNet128dClient,
-        "Facenet512": FaceNet.FaceNet512dClient,
+        "Facenet": Facenet.FaceNet128dClient,
+        "Facenet512": Facenet.FaceNet512dClient,
         "DeepFace": FbDeepFace.DeepFaceClient,
         "DeepID": DeepID.DeepIdClient,
         "Dlib": Dlib.DlibClient,