c-l/immich
1
0
Fork 0
mirror of https://github.com/immich-app/immich.git synced 2024-12-29 15:11:58 +00:00
Code Releases Activity

change log message

Browse source
This commit is contained in:
mertalev 2024-06-08 21:22:38 -04:00
parent 717961ce7b
commit fb4fe5d40b
No known key found for this signature in database
GPG key ID: 9181CD92C0A1C5E3
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/app/models/facial_recognition/recognition.py
View file

@ -32,7 +32,7 @@ class FaceRecognizer(InferenceModel):
def _load(self) -> ModelSession: def _load(self) -> ModelSession:
session = self._make_session(self.model_path) session = self._make_session(self.model_path)
if isinstance(session, ort.InferenceSession) and not ort_has_batch_dim(session): if isinstance(session, ort.InferenceSession) and not ort_has_batch_dim(session):
log.debug(f"Adding batch dimension to model {self.model_path}") log.info(f"Adding batch dimension to recognition model {self.model_name}")
ort_add_batch_dim(self.model_path, self.model_path) ort_add_batch_dim(self.model_path, self.model_path)
session = self._make_session(self.model_path) session = self._make_session(self.model_path)
self.model = ArcFaceONNX( self.model = ArcFaceONNX(