fix(ml): re-enable ARMNN integration (#7053)

2024-02-12 19:29:55 +01:00
parent dca1bd22df
commit 49c378d706
2 changed files with 6 additions and 2 deletions
@@ -28,7 +28,10 @@ class FaceRecognizer(InferenceModel):

    def _load(self) -> None:
        self.det_model = RetinaFace(session=self._make_session(self.det_file))
-        self.rec_model = ArcFaceONNX(self.rec_file.as_posix(), session=self._make_session(self.rec_file))
+        self.rec_model = ArcFaceONNX(
+            self.rec_file.with_suffix(".onnx").as_posix(),
+            session=self._make_session(self.rec_file),
+        )

        self.det_model.prepare(
            ctx_id=0,