From 9852bbe48236dfa79689e15cdc047297a4c07de4 Mon Sep 17 00:00:00 2001 From: Rodrigo Andrade Date: Sun, 8 May 2022 21:19:52 -0300 Subject: [PATCH] added sface --- deepface/DeepFace.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) 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,