diff --git a/deepface/models/face_detection/Ssd.py b/deepface/models/face_detection/Ssd.py index 2ccc3c5..4250888 100644 --- a/deepface/models/face_detection/Ssd.py +++ b/deepface/models/face_detection/Ssd.py @@ -107,7 +107,7 @@ class SsdClient(Detector): 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]