From 387ded5c66d5f713606bab3fa166d742e509f0c9 Mon Sep 17 00:00:00 2001
From: Sefik Ilkin Serengil <serengil@gmail.com>
Date: Fri, 11 Oct 2024 19:44:11 +0100
Subject: [PATCH] min detection confidence score retrieved from env vars

---
 deepface/models/face_detection/Yolo.py | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/deepface/models/face_detection/Yolo.py b/deepface/models/face_detection/Yolo.py
index a0211ed..77dd09b 100644
--- a/deepface/models/face_detection/Yolo.py
+++ b/deepface/models/face_detection/Yolo.py
@@ -1,4 +1,5 @@
 # built-in dependencies
+import os
 from typing import Any, List
 
 # 3rd party dependencies
@@ -58,7 +59,12 @@ class YoloClient(Detector):
         resp = []
 
         # Detect faces
-        results = self.model.predict(img, verbose=False, show=False, conf=0.25)[0]
+        results = self.model.predict(
+            img,
+            verbose=False,
+            show=False,
+            conf=float(os.getenv("YOLO_MIN_DETECTION_CONFIDENCE", "0.25")),
+        )[0]
 
         # For each face, extract the bounding box, the landmarks and confidence
         for result in results: