From c25f14af104140ac70fb0af35f544813cd52d47c Mon Sep 17 00:00:00 2001 From: roberto-corno-nttdata Date: Mon, 9 Dec 2024 16:04:08 +0100 Subject: [PATCH] fix: yolo input_shape --- deepface/models/facial_recognition/Yolo.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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)