diff --git a/deepface/models/facial_recognition/Yolo.py b/deepface/models/facial_recognition/Yolo.py index 1e08218..b87f8fc 100644 --- a/deepface/models/facial_recognition/Yolo.py +++ b/deepface/models/facial_recognition/Yolo.py @@ -37,7 +37,7 @@ class YoloFacialRecognitionClient(FacialRecognition): def __init__(self, model: YoloModel): super().__init__() self.model_name = "Yolo" - self.input_shape = (224, 224) + self.input_shape = (640, 640) self.output_shape = 512 self.model = self.build_model(model)