From 5f1c51350ae884ad7ea879abe725a0308459150f Mon Sep 17 00:00:00 2001 From: tyasnk Date: Thu, 10 Oct 2024 16:58:58 +0700 Subject: [PATCH] Feat: Configurable Mediapipe parameters --- deepface/models/face_detection/MediaPipe.py | 11 +++++++---- 1 file changed, 7 insertions(+), 4 deletions(-) diff --git a/deepface/models/face_detection/MediaPipe.py b/deepface/models/face_detection/MediaPipe.py index 9b0f9d6..099d970 100644 --- a/deepface/models/face_detection/MediaPipe.py +++ b/deepface/models/face_detection/MediaPipe.py @@ -1,4 +1,5 @@ # built-in dependencies +import os from typing import Any, List # 3rd party dependencies @@ -7,8 +8,6 @@ import numpy as np # project dependencies from deepface.models.Detector import Detector, FacialAreaRegion -import os - class MediaPipeClient(Detector): """ @@ -34,9 +33,13 @@ class MediaPipeClient(Detector): ) from e mp_face_detection = mp.solutions.face_detection + + min_detection_confidence = float(os.environ.get("MEDIAPIPE_MIN_DETECTION_CONFIDENCE", 0.7)) + model_selection = float(os.environ.get("MEDIAPIPE_MODEL_SELECTION", 0)) + 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) + min_detection_confidence=min_detection_confidence, + model_selection=model_selection ) return face_detection