From cef09cbdca3f2cc690dbd62c5c41c4b32aef22ee Mon Sep 17 00:00:00 2001 From: Shivam Singhal Date: Thu, 29 Aug 2024 12:06:24 +0530 Subject: [PATCH] fix: fp32 changes --- deepface/models/facial_recognition/Facenet.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/deepface/models/facial_recognition/Facenet.py b/deepface/models/facial_recognition/Facenet.py index da14027..da67af6 100644 --- a/deepface/models/facial_recognition/Facenet.py +++ b/deepface/models/facial_recognition/Facenet.py @@ -83,7 +83,7 @@ class FaceNet512dONNXClient(FacialRecognition): def forward(self, img: np.ndarray) -> List[float]: 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.astype(np.float16)}) + result = self.model.run([output_name], {input_name: img}) return result[0][0].tolist()