github/deepface
1
1
Fork 0
mirror of https://github.com/serengil/deepface.git synced 2025-06-07 03:55:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: np array

Browse Source
This commit is contained in:
Shivam Singhal 2024-08-27 19:45:51 +05:30
parent 0037527fc3
commit 85da7a6320
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
deepface/models/facial_recognition/Facenet.py
View File

@ -84,7 +84,9 @@ class FaceNet512dONNXClient(FacialRecognition):
input_name = self.model.get_inputs()[0].name input_name = self.model.get_inputs()[0].name
output_name = self.model.get_outputs()[0].name output_name = self.model.get_outputs()[0].name
result = self.model.run([output_name], {input_name: img}) 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): def scaling(x, scale):