diff --git a/machine-learning/immich_ml/models/facial_recognition/detection.py b/machine-learning/immich_ml/models/facial_recognition/detection.py index cadeee4ccc..5e5015574c 100644 --- a/machine-learning/immich_ml/models/facial_recognition/detection.py +++ b/machine-learning/immich_ml/models/facial_recognition/detection.py @@ -20,7 +20,7 @@ class FaceDetector(InferenceModel): def _load(self) -> ModelSession: session = self._make_session(self.model_path) self.model = RetinaFace(session=session) - self.model.prepare(ctx_id=0, det_thresh=self.min_score, input_size=(256, 256)) + self.model.prepare(ctx_id=0, det_thresh=self.min_score, input_size=(640, 640)) return session