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
import math
import numpy as np
@ -13,7 +13,8 @@ def build_model(detector_backend):
'ssd': SsdWrapper.build_model,
'dlib': DlibWrapper.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():
@ -50,7 +51,8 @@ def detect_faces(face_detector, detector_backend, img, align = True):
'ssd': SsdWrapper.detect_face,
'dlib': DlibWrapper.detect_face,
'mtcnn': MtcnnWrapper.detect_face,
'retinaface': RetinaFaceWrapper.detect_face
'retinaface': RetinaFaceWrapper.detect_face,
'mediapipe': MediapipeWrapper.detect_face
}
detect_face = backends.get(detector_backend)