From fdc8f91b182defe0179fdcf812daeb8d18810c08 Mon Sep 17 00:00:00 2001
From: mertalev <101130780+mertalev@users.noreply.github.com>
Date: Tue, 13 May 2025 23:45:59 -0400
Subject: [PATCH] revert image size change

---
 .../immich_ml/models/facial_recognition/detection.py            | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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