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

Feat: Configurable Mediapipe parameters

Browse Source
This commit is contained in:
tyasnk 2024-10-10 16:58:58 +07:00
parent 49c041945a
commit 5f1c51350a
1 changed files with 7 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -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