From e9d980fdd46cf020f119d320695ea05791f31513 Mon Sep 17 00:00:00 2001 From: Shivam Singhal Date: Tue, 27 Aug 2024 21:56:34 +0530 Subject: [PATCH] feat: any change --- deepface/models/facial_recognition/Facenet.py | 6 ++---- tests/test_verify.py | 4 ++-- 2 files changed, 4 insertions(+), 6 deletions(-) diff --git a/deepface/models/facial_recognition/Facenet.py b/deepface/models/facial_recognition/Facenet.py index 7529de7..9d225c9 100644 --- a/deepface/models/facial_recognition/Facenet.py +++ b/deepface/models/facial_recognition/Facenet.py @@ -84,9 +84,7 @@ 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}) - if isinstance(result[0][0], np.ndarray): - return result[0][0].tolist() - return list(result[0][0]) + return result[0][0].tolist() def scaling(x, scale): @@ -1741,7 +1739,7 @@ def load_facenet512d_onnx_model( """ Download Facenet512d ONNX model weights and load Returns: - model (Model) + model (Any) """ try: import onnxruntime as ort diff --git a/tests/test_verify.py b/tests/test_verify.py index c1a80e4..39d8af5 100644 --- a/tests/test_verify.py +++ b/tests/test_verify.py @@ -8,9 +8,9 @@ from deepface.commons.logger import Logger logger = Logger() -models = ["VGG-Face", "Facenet", "Facenet512", "ArcFace", "GhostFaceNet"] +models = ["Facenet512ONNX"] metrics = ["cosine", "euclidean", "euclidean_l2"] -detectors = ["opencv", "mtcnn"] +detectors = ["opencv"] def test_different_facial_recognition_models():