Cutlery/immich
1
0
Fork 1
mirror of https://github.com/immich-app/immich.git synced 2025-05-24 01:12:58 -04:00
Code Issues Packages Projects Releases Wiki Activity

revert image size change

Browse Source
This commit is contained in:
mertalev 2025-05-13 23:45:59 -04:00
parent 016a760dda
commit fdc8f91b18
No known key found for this signature in database
GPG Key ID: DF6ABC77AAD98C95
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
machine-learning/immich_ml/models/facial_recognition/detection.py
View File

@ -20,7 +20,7 @@ class FaceDetector(InferenceModel):
def _load(self) -> ModelSession: def _load(self) -> ModelSession:
session = self._make_session(self.model_path) session = self._make_session(self.model_path)
self.model = RetinaFace(session=session) 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 return session