From ef0b71a0af536c25c44501f10d5c47f751453fb9 Mon Sep 17 00:00:00 2001 From: kremnik Date: Sun, 1 Sep 2024 17:41:22 +0300 Subject: [PATCH] Change index to label in ssd detector --- deepface/models/face_detection/Ssd.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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]