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

FaceNet to Facenet

Browse Source
This commit is contained in:
Sefik Ilkin Serengil 2024-01-21 14:01:11 +00:00
parent 0d6aa10b82
commit c9a7db8815
1 changed files with 3 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
deepface/modules/modeling.py
View File

@ -2,16 +2,7 @@
from typing import Any from typing import Any
# project dependencies # project dependencies
from deepface.basemodels import ( from deepface.basemodels import VGGFace, OpenFace, FbDeepFace, DeepID, ArcFace, SFace, Dlib, Facenet
VGGFace,
OpenFace,
FbDeepFace,
DeepID,
ArcFace,
SFace,
Dlib,
FaceNet,
)
from deepface.extendedmodels import Age, Gender, Race, Emotion from deepface.extendedmodels import Age, Gender, Race, Emotion
@ -33,8 +24,8 @@ def build_model(model_name: str) -> Any:
models = { models = {
"VGG-Face": VGGFace.VggFaceClient, "VGG-Face": VGGFace.VggFaceClient,
"OpenFace": OpenFace.OpenFaceClient, "OpenFace": OpenFace.OpenFaceClient,
"Facenet": FaceNet.FaceNet128dClient, "Facenet": Facenet.FaceNet128dClient,
"Facenet512": FaceNet.FaceNet512dClient, "Facenet512": Facenet.FaceNet512dClient,
"DeepFace": FbDeepFace.DeepFaceClient, "DeepFace": FbDeepFace.DeepFaceClient,
"DeepID": DeepID.DeepIdClient, "DeepID": DeepID.DeepIdClient,
"Dlib": Dlib.DlibClient, "Dlib": Dlib.DlibClient,