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

Set model detection to int

Browse Source
This commit is contained in:
tyasnk 2024-10-10 17:03:38 +07:00
parent 5f1c51350a
commit 8766a412ec
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -35,7 +35,7 @@ class MediaPipeClient(Detector):
mp_face_detection = mp.solutions.face_detection mp_face_detection = mp.solutions.face_detection
min_detection_confidence = float(os.environ.get("MEDIAPIPE_MIN_DETECTION_CONFIDENCE", 0.7)) min_detection_confidence = float(os.environ.get("MEDIAPIPE_MIN_DETECTION_CONFIDENCE", 0.7))
model_selection = float(os.environ.get("MEDIAPIPE_MODEL_SELECTION", 0)) model_selection = int(os.environ.get("MEDIAPIPE_MODEL_SELECTION", 0))
face_detection = mp_face_detection.FaceDetection( face_detection = mp_face_detection.FaceDetection(
min_detection_confidence=min_detection_confidence, min_detection_confidence=min_detection_confidence,