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

use preprocessing from opencv

Browse Source
This commit is contained in:
Rodrigo Andrade 2022-05-08 21:27:42 -03:00
parent 9852bbe482
commit a5ecdc1d8c
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
deepface/basemodels/sface_opencv_wrapper.py
View File

@ -1,5 +1,6 @@
import os
import numpy as np
import cv2 as cv
import gdown
@ -33,11 +34,11 @@ class SFace:
def predict(self, image, bbox=None, **kwargs):
# Preprocess
# print(image.max())
# input_blob = self._preprocess(image, bbox)
image = (image[0] * 255).astype(np.uint8) # revert the iamge to original format and preprocess using the model
input_blob = self._preprocess(image, bbox)
# Forward
features = self._model.feature(image[0])
features = self._model.feature(input_blob)
return features