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

Update FaceDetector.py

Browse Source
This commit is contained in:
blueprintparadise 2021-09-28 16:28:20 +05:30 committed by GitHub
parent ae10f5a352
commit f07f278d22
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
deepface/detectors/FaceDetector.py
View File

@ -1,4 +1,4 @@
from deepface.detectors import OpenCvWrapper, SsdWrapper, DlibWrapper, MtcnnWrapper, RetinaFaceWrapper from deepface.detectors import OpenCvWrapper, SsdWrapper, DlibWrapper, MtcnnWrapper, RetinaFaceWrapper,MediapipeWrapper
from PIL import Image from PIL import Image
import math import math
import numpy as np import numpy as np
@ -13,7 +13,8 @@ def build_model(detector_backend):
'ssd': SsdWrapper.build_model, 'ssd': SsdWrapper.build_model,
'dlib': DlibWrapper.build_model, 'dlib': DlibWrapper.build_model,
'mtcnn': MtcnnWrapper.build_model, 'mtcnn': MtcnnWrapper.build_model,
'retinaface': RetinaFaceWrapper.build_model 'retinaface': RetinaFaceWrapper.build_model,
'mediapipe': MediapipeWrapper.build_model
} }
if not "face_detector_obj" in globals(): if not "face_detector_obj" in globals():
@ -50,7 +51,8 @@ def detect_faces(face_detector, detector_backend, img, align = True):
'ssd': SsdWrapper.detect_face, 'ssd': SsdWrapper.detect_face,
'dlib': DlibWrapper.detect_face, 'dlib': DlibWrapper.detect_face,
'mtcnn': MtcnnWrapper.detect_face, 'mtcnn': MtcnnWrapper.detect_face,
'retinaface': RetinaFaceWrapper.detect_face 'retinaface': RetinaFaceWrapper.detect_face,
'mediapipe': MediapipeWrapper.detect_face
} }
detect_face = backends.get(detector_backend) detect_face = backends.get(detector_backend)