From 8e5e42d4d3bee46546dd5d73c771e1dda1d7a600 Mon Sep 17 00:00:00 2001 From: tyasnk Date: Thu, 3 Oct 2024 10:39:27 +0700 Subject: [PATCH] Mediapipe paramater now configurable by env variable --- deepface/models/face_detection/MediaPipe.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/deepface/models/face_detection/MediaPipe.py b/deepface/models/face_detection/MediaPipe.py index 4358516..9b0f9d6 100644 --- a/deepface/models/face_detection/MediaPipe.py +++ b/deepface/models/face_detection/MediaPipe.py @@ -7,6 +7,8 @@ import numpy as np # project dependencies from deepface.models.Detector import Detector, FacialAreaRegion +import os + class MediaPipeClient(Detector): """ @@ -32,7 +34,10 @@ class MediaPipeClient(Detector): ) from e mp_face_detection = mp.solutions.face_detection - face_detection = mp_face_detection.FaceDetection(min_detection_confidence=0.5, model_selection=1) + face_detection = mp_face_detection.FaceDetection( + min_detection_confidence=os.getenv("MEDIAPIPE_MIN_DETECTION_CONFIDENCE", 0.5), + model_selection=os.getenv("MEDIAPIPE_MODEL_SELECTION", 1) + ) return face_detection def detect_faces(self, img: np.ndarray) -> List[FacialAreaRegion]: