github/deepface
1
1
Fork 0
mirror of https://github.com/serengil/deepface.git synced 2025-06-07 20:15:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

detector as a global variable

Browse Source
This commit is contained in:
Sefik Ilkin Serengil 2021-06-24 00:00:20 +03:00
parent d18905a8ac
commit 9f31012ef4
1 changed files with 10 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
deepface/detectors/FaceDetector.py
View File

@ -6,6 +6,8 @@ from deepface.commons import distance
def build_model(detector_backend): def build_model(detector_backend):
global face_detector_obj, face_detector_label
backends = { backends = {
'opencv': OpenCvWrapper.build_model, 'opencv': OpenCvWrapper.build_model,
'ssd': SsdWrapper.build_model, 'ssd': SsdWrapper.build_model,
@ -14,14 +16,16 @@ def build_model(detector_backend):
'retinaface': RetinaFaceWrapper.build_model 'retinaface': RetinaFaceWrapper.build_model
} }
face_detector = backends.get(detector_backend) if not "face_detector_obj" in globals() or face_detector_label != detector_backend:
face_detector_obj = backends.get(detector_backend)
face_detector_label = detector_backend
if face_detector: if face_detector_obj:
face_detector = face_detector() face_detector_obj = face_detector_obj()
else: else:
raise ValueError("invalid detector_backend passed - " + detector_backend) raise ValueError("invalid detector_backend passed - " + detector_backend)
return face_detector return face_detector_obj
def detect_face(face_detector, detector_backend, img, align = True): def detect_face(face_detector, detector_backend, img, align = True):