diff --git a/deepface/DeepFace.py b/deepface/DeepFace.py index dade599..543bcb2 100644 --- a/deepface/DeepFace.py +++ b/deepface/DeepFace.py @@ -12,7 +12,8 @@ import pandas as pd from tqdm import tqdm 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.commons import functions, realtime, distance as dst @@ -46,6 +47,7 @@ def build_model(model_name): 'DeepID': DeepID.loadModel, 'Dlib': DlibWrapper.loadModel, 'ArcFace': ArcFace.loadModel, + 'SFace': sface_opencv_wrapper.load_model, 'Emotion': Emotion.loadModel, 'Age': Age.loadModel, 'Gender': Gender.loadModel,