From f07f278d223922c9a9fcfc96eb63bd7c63d54f48 Mon Sep 17 00:00:00 2001 From: blueprintparadise Date: Tue, 28 Sep 2021 16:28:20 +0530 Subject: [PATCH] Update FaceDetector.py --- deepface/detectors/FaceDetector.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/deepface/detectors/FaceDetector.py b/deepface/detectors/FaceDetector.py index ad301a4..971670f 100644 --- a/deepface/detectors/FaceDetector.py +++ b/deepface/detectors/FaceDetector.py @@ -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)