Merge pull request serengil#864 from haddyadnan/face-confidence

Include detected face confidence
Neil-Huang · Oct 18, 2023 · 3e1f48c · 3e1f48c
2 parents fb8924e + 0a389de
commit 3e1f48c
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/deepface/DeepFace.py b/deepface/DeepFace.py
@@ -264,6 +264,7 @@ def analyze(
  {
  "region": {'x': 230, 'y': 120, 'w': 36, 'h': 45},
  "age": 28.66,
+ 'face_confidence': 0.9993908405303955,
  "dominant_gender": "Woman",
  "gender": {
  'Woman': 99.99407529830933,
@@ -335,7 +336,7 @@ def analyze(
  align=align,
  )
 
- for img_content, img_region, _ in img_objs:
+ for img_content, img_region, img_confidence in img_objs:
  if img_content.shape[0] > 0 and img_content.shape[1] > 0:
  obj = {}
  # facial attribute analysis
@@ -390,6 +391,8 @@ def analyze(
  # -----------------------------
  # mention facial areas
  obj["region"] = img_region
+ # include image confidence
+ obj["face_confidence"] = img_confidence
 
  resp_objects.append(obj)