initialize detector in represent

This commit is contained in:
Sefik Ilkin Serengil 2021-04-29 21:19:49 +03:00
parent 91aa422677
commit 73d1950133

View File

@ -43,6 +43,7 @@ def build_model(model_name):
'DeepID': DeepID.loadModel, 'DeepID': DeepID.loadModel,
'Dlib': DlibWrapper.loadModel, 'Dlib': DlibWrapper.loadModel,
'ArcFace': ArcFace.loadModel, 'ArcFace': ArcFace.loadModel,
'Emotion': Emotion.loadModel, 'Emotion': Emotion.loadModel,
'Age': Age.loadModel, 'Age': Age.loadModel,
'Gender': Gender.loadModel, 'Gender': Gender.loadModel,
@ -725,6 +726,8 @@ def represent(img_path, model_name = 'VGG-Face', model = None, enforce_detection
if model is None: if model is None:
model = build_model(model_name) model = build_model(model_name)
functions.initialize_detector(detector_backend = detector_backend)
#--------------------------------- #---------------------------------
#decide input shape #decide input shape