Merge pull request #1330 from kremnik/ssdpandas

serengil · Sep 1, 2024 · d3200c2 · d3200c2
2 parents a80d3e8 + ef0b71a
commit d3200c2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/deepface/models/face_detection/Ssd.py b/deepface/models/face_detection/Ssd.py
@@ -107,7 +107,7 @@ class ssd_labels(IntEnum):
 
         resp = []
         for face in faces:
-            confidence = float(face[2])
+            confidence = float(face[ssd_labels.confidence])
             x, y, w, h = map(int, face[margins])
             detected_face = img[y : y + h, x : x + w]