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

clean code in detectors interface

Browse Source
This commit is contained in:
Sefik Ilkin Serengil 2021-04-28 23:03:15 +03:00
parent b898f97b38
commit 0bdc341ce7
2 changed files with 20 additions and 26 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

46
deepface/detectors/FaceDetector.py
View File

@ -6,21 +6,18 @@ from deepface.commons import distance
def build_model(detector_backend):
if detector_backend == 'opencv':
face_detector = OpenCvWrapper.build_model()
backends = {
'opencv': OpenCvWrapper.build_model,
'ssd': SsdWrapper.build_model,
'dlib': DlibWrapper.build_model,
'mtcnn': MtcnnWrapper.build_model,
'retinaface': RetinaFaceWrapper.build_model
}
elif detector_backend == 'ssd':
face_detector = SsdWrapper.build_model()
elif detector_backend == 'dlib':
face_detector = DlibWrapper.build_model()
elif detector_backend == 'mtcnn':
face_detector = MtcnnWrapper.build_model()
elif detector_backend == 'retinaface':
face_detector = RetinaFaceWrapper.build_model()
face_detector = backends.get(detector_backend)
if face_detector:
face_detector = face_detector()
else:
raise ValueError("invalid detector_backend passed - " + detector_backend)
@ -28,21 +25,18 @@ def build_model(detector_backend):
def detect_face(face_detector, detector_backend, img):
if detector_backend == 'opencv':
face, region = OpenCvWrapper.detect_face(face_detector, img)
backends = {
'opencv': OpenCvWrapper.detect_face,
'ssd': SsdWrapper.detect_face,
'dlib': DlibWrapper.detect_face,
'mtcnn': MtcnnWrapper.detect_face,
'retinaface': RetinaFaceWrapper.detect_face
}
elif detector_backend == 'ssd':
face, region = SsdWrapper.detect_face(face_detector, img)
elif detector_backend == 'dlib':
face, region = DlibWrapper.detect_face(face_detector, img)
elif detector_backend == 'mtcnn':
face, region = MtcnnWrapper.detect_face(face_detector, img)
elif detector_backend == 'retinaface':
face, region = RetinaFaceWrapper.detect_face(face_detector, img)
detect_face = backends.get(detector_backend)
if detect_face:
face, region = detect_face(face_detector, img)
else:
raise ValueError("invalid detector_backend passed - " + detector_backend)

0
deepface/detectors/__init__.py Normal file
View File