diff --git a/deepface/models/face_detection/Ssd.py b/deepface/models/face_detection/Ssd.py index c388a5992..cc02fb75b 100644 --- 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]