fix: added input_shape to YoloFacialRecognitionClient

deepface/models/facial_recognition/Yolo.py

@@ -35,10 +35,11 @@ class YoloModel(Enum):
 
 class YoloFacialRecognitionClient(FacialRecognition):
     def __init__(self, model: YoloModel):
-        super().__init__(model)
+        super().__init__()
         self.model_name = "Yolo"
-        self.input_shape = None
+        self.input_shape = (224, 224)
         self.output_shape = 512
+        self.model = self.build_model(model)
 
     def build_model(self, model: YoloModel) -> Any:
         """
@@ -64,7 +65,7 @@ def build_model(self, model: YoloModel) -> Any:
         return YOLO(weight_file)
 
     def forward(self, img: np.ndarray) -> List[float]:
-        return self.model.embed(img)[0].tolist()
+        return self.model.embed(np.squeeze(img, axis=0))[0].tolist()
 
 
 class YoloFacialRecognitionClientV8n(YoloFacialRecognitionClient):

deepface/modules/representation.py

@@ -122,12 +122,11 @@ def represent(
         confidence = img_obj["confidence"]
 
         # resize to expected shape of ml model
-        if target_size is not None:
-            img = preprocessing.resize_image(
-                img=img,
-                # thanks to DeepId (!)
-                target_size=(target_size[1], target_size[0]),
-            )
+        img = preprocessing.resize_image(
+            img=img,
+            # thanks to DeepId (!)
+            target_size=(target_size[1], target_size[0]),
+        )
 
         # custom normalization
         img = preprocessing.normalize_input(img=img, normalization=normalization)