From a5ecdc1d8cf854a8725e6ee8f539af400899bf0b Mon Sep 17 00:00:00 2001 From: Rodrigo Andrade Date: Sun, 8 May 2022 21:27:42 -0300 Subject: [PATCH] use preprocessing from opencv --- deepface/basemodels/sface_opencv_wrapper.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/deepface/basemodels/sface_opencv_wrapper.py b/deepface/basemodels/sface_opencv_wrapper.py index 3c6ea0e..d28444b 100644 --- a/deepface/basemodels/sface_opencv_wrapper.py +++ b/deepface/basemodels/sface_opencv_wrapper.py @@ -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