helpy4/immich
1
0
Fork 0
mirror of https://github.com/immich-app/immich synced 2025-11-07 17:27:20 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

feat(ml): ML on Rockchip NPUs (#15241)

Browse source
This commit is contained in:
Yoni Yang 2025-03-18 00:04:08 +08:00 • committed by GitHub
parent 1e184a70f1
commit 14c3b99c0f
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
43 changed files with 2417 additions and 4726 deletions
Download patch file Download diff file Expand all files Collapse all files

2
machine-learning/app/models/facial_recognition/recognition.py
View file

@ -31,7 +31,7 @@ class FaceRecognizer(InferenceModel):
self._add_batch_axis(self.model_path)
session = self._make_session(self.model_path)
self.model = ArcFaceONNX(
self.model_path.with_suffix(".onnx").as_posix(),
self.model_path_for_format(ModelFormat.ONNX).as_posix(),
session=session,
)
return session