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
# 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,