cjones/immich
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

change log message

Browse Source
This commit is contained in:
mertalev
2024-06-08 21:22:38 -04:00
parent 717961ce7b
commit fb4fe5d40b
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
machine-learning/app/models/facial_recognition/recognition.py
+1 -1
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(