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

fix: fp32 changes

Browse Source
This commit is contained in:
Shivam Singhal 2024-08-29 12:06:24 +05:30
parent 9889356b7b
commit cef09cbdca
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -83,7 +83,7 @@ class FaceNet512dONNXClient(FacialRecognition):
def forward(self, img: np.ndarray) -> List[float]: def forward(self, img: np.ndarray) -> List[float]:
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.astype(np.float16)}) result = self.model.run([output_name], {input_name: img})
return result[0][0].tolist() return result[0][0].tolist()