github/deepface
1
1
Fork 0
mirror of https://github.com/serengil/deepface.git synced 2025-07-24 02:40:00 +00:00
Code Issues Packages Projects Releases Wiki Activity

ghostface added in portfolio

Browse Source
This commit is contained in:
Sefik Ilkin Serengil 2024-03-16 09:41:35 +00:00
parent 5825e2aa88
commit 808936dc24
1 changed files with 12 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
deepface/modules/modeling.py
View File

@ -2,7 +2,17 @@
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,
GhostFaceNet
)
from deepface.extendedmodels import Age, Gender, Race, Emotion
@ -31,6 +41,7 @@ def build_model(model_name: str) -> Any:
"Dlib": Dlib.DlibClient,
"ArcFace": ArcFace.ArcFaceClient,
"SFace": SFace.SFaceClient,
"GhostFaceNet": GhostFaceNet.GhostFaceNetClient,
"Emotion": Emotion.EmotionClient,
"Age": Age.ApparentAgeClient,
"Gender": Gender.GenderClient,