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

Mediapipe paramater now configurable by env variable

Browse Source
This commit is contained in:
tyasnk 2024-10-03 10:39:27 +07:00
parent c4a49fff98
commit 8e5e42d4d3
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
deepface/models/face_detection/MediaPipe.py
View File

@ -7,6 +7,8 @@ import numpy as np
# project dependencies # project dependencies
from deepface.models.Detector import Detector, FacialAreaRegion from deepface.models.Detector import Detector, FacialAreaRegion
import os
class MediaPipeClient(Detector): class MediaPipeClient(Detector):
""" """
@ -32,7 +34,10 @@ class MediaPipeClient(Detector):
) from e ) from e
mp_face_detection = mp.solutions.face_detection 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 return face_detection
def detect_faces(self, img: np.ndarray) -> List[FacialAreaRegion]: def detect_faces(self, img: np.ndarray) -> List[FacialAreaRegion]: