Update Ssd.py

ssd's confidence score should be float not numpy
serengil · Aug 30, 2024 · 5e261e7 · 5e261e7
1 parent 14158d3
commit 5e261e7
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
@@ -105,7 +105,7 @@ class ssd_labels(IntEnum):
 
         resp = []
         for face in faces:
-            confidence = face[2]
+            confidence = float(face[2])
             x, y, w, h = map(int, face[margins])
             detected_face = img[y : y + h, x : x + w]