Change index to label in ssd detector

This commit is contained in:
kremnik 2024-09-01 17:41:22 +03:00
parent 2feb703f96
commit ef0b71a0af

View File

@ -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]