From 85da7a6320f860eda68a8462f6c7aba907651b1e Mon Sep 17 00:00:00 2001 From: Shivam Singhal Date: Tue, 27 Aug 2024 19:45:51 +0530 Subject: [PATCH] feat: np array --- deepface/models/facial_recognition/Facenet.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/deepface/models/facial_recognition/Facenet.py b/deepface/models/facial_recognition/Facenet.py index e619392..683cc3b 100644 --- a/deepface/models/facial_recognition/Facenet.py +++ b/deepface/models/facial_recognition/Facenet.py @@ -84,7 +84,9 @@ class FaceNet512dONNXClient(FacialRecognition): input_name = self.model.get_inputs()[0].name output_name = self.model.get_outputs()[0].name result = self.model.run([output_name], {input_name: img}) - return result[0][0] + if isinstance(result[0][0], np.ndarray): + return result[0][0].tolist() + return list(result[0][0]) def scaling(x, scale):