Backupppert/immich
1
1
Fork 0
mirror of https://github.com/immich-app/immich.git synced 2025-07-15 20:38:26 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions6

feat(ml): conditionally download .armnn models ()

Browse source
This commit is contained in:
Mert 2024-01-28 10:31:59 -05:00 • committed by GitHub
parent fa0913120d
commit a84b6f5fb1
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
5 changed files with 127 additions and 38 deletions
machine-learning/app/models
facial_recognition.py

4
machine-learning/app/models/facial_recognition.py
View file

@ -77,11 +77,11 @@ class FaceRecognizer(InferenceModel):
@property
def det_file(self) -> Path:
return self.cache_dir / "detection" / "model.onnx"
return self.cache_dir / "detection" / f"model.{self.preferred_runtime}"
@property
def rec_file(self) -> Path:
return self.cache_dir / "recognition" / "model.onnx"
return self.cache_dir / "recognition" / f"model.{self.preferred_runtime}"
def configure(self, **model_kwargs: Any) -> None:
self.det_model.det_thresh = model_kwargs.pop("minScore", self.det_model.det_thresh)