mirror of
https://github.com/serengil/deepface.git
synced 2025-06-12 14:27:09 +00:00
added sface
This commit is contained in:
parent
f74e944dc6
commit
9852bbe482
@ -12,7 +12,8 @@ import pandas as pd
|
|||||||
from tqdm import tqdm
|
from tqdm import tqdm
|
||||||
import pickle
|
import pickle
|
||||||
|
|
||||||
from deepface.basemodels import VGGFace, OpenFace, Facenet, Facenet512, FbDeepFace, DeepID, DlibWrapper, ArcFace, Boosting
|
from deepface.basemodels import VGGFace, OpenFace, Facenet, Facenet512, FbDeepFace, DeepID, DlibWrapper, ArcFace, \
|
||||||
|
Boosting, sface_opencv_wrapper
|
||||||
from deepface.extendedmodels import Age, Gender, Race, Emotion
|
from deepface.extendedmodels import Age, Gender, Race, Emotion
|
||||||
from deepface.commons import functions, realtime, distance as dst
|
from deepface.commons import functions, realtime, distance as dst
|
||||||
|
|
||||||
@ -46,6 +47,7 @@ def build_model(model_name):
|
|||||||
'DeepID': DeepID.loadModel,
|
'DeepID': DeepID.loadModel,
|
||||||
'Dlib': DlibWrapper.loadModel,
|
'Dlib': DlibWrapper.loadModel,
|
||||||
'ArcFace': ArcFace.loadModel,
|
'ArcFace': ArcFace.loadModel,
|
||||||
|
'SFace': sface_opencv_wrapper.load_model,
|
||||||
'Emotion': Emotion.loadModel,
|
'Emotion': Emotion.loadModel,
|
||||||
'Age': Age.loadModel,
|
'Age': Age.loadModel,
|
||||||
'Gender': Gender.loadModel,
|
'Gender': Gender.loadModel,
|
||||||
|
Loading…
x
Reference in New Issue
Block a user