diff --git a/models/face_recognition_sface/sface.py b/models/face_recognition_sface/sface.py index 84e2bda2..98acf612 100644 --- a/models/face_recognition_sface/sface.py +++ b/models/face_recognition_sface/sface.py @@ -47,9 +47,12 @@ def setTarget(self, targetId): target_id=self._targetId) def _preprocess(self, image, bbox): - return self._model.alignCrop(image, bbox) + if bbox is None: + return image + else: + return self._model.alignCrop(image, bbox) - def infer(self, image, bbox): + def infer(self, image, bbox=None): # Preprocess inputBlob = self._preprocess(image, bbox) @@ -66,4 +69,5 @@ def match(self, image1, face1, image2, face2): return 1 if cosine_score >= self._threshold_cosine else 0 else: # NORM_L2 norml2_distance = self._model.match(feature1, feature2, self._disType) - return 1 if norml2_distance <= self._threshold_norml2 else 0 \ No newline at end of file + return 1 if norml2_distance <= self._threshold_norml2 else 0 +