From 6f335800e9db20ba413d976a29eda78d5e033759 Mon Sep 17 00:00:00 2001 From: Nicolas Mowen Date: Thu, 10 Oct 2024 10:10:28 -0600 Subject: [PATCH] Add debugging logs for more info --- frigate/embeddings/embeddings.py | 2 ++ frigate/embeddings/functions/onnx.py | 6 +++++- 2 files changed, 7 insertions(+), 1 deletion(-) diff --git a/frigate/embeddings/embeddings.py b/frigate/embeddings/embeddings.py index dda4d95fd..35125aabc 100644 --- a/frigate/embeddings/embeddings.py +++ b/frigate/embeddings/embeddings.py @@ -168,7 +168,9 @@ class Embeddings: logger.info("Indexing tracked object embeddings...") self._drop_tables() + logger.debug("Dropped embeddings tables.") self._create_tables() + logger.debug("Created embeddings tables.") st = time.time() totals = { diff --git a/frigate/embeddings/functions/onnx.py b/frigate/embeddings/functions/onnx.py index 08901b6a2..448f9f75b 100644 --- a/frigate/embeddings/functions/onnx.py +++ b/frigate/embeddings/functions/onnx.py @@ -110,6 +110,7 @@ class GenericONNXEmbedding: self.session = self._load_model( os.path.join(self.download_path, self.model_file) ) + logger.debug("successfully loaded model.") def _load_tokenizer(self): tokenizer_path = os.path.join(f"{MODEL_CACHE_DIR}/{self.model_name}/tokenizer") @@ -125,8 +126,11 @@ class GenericONNXEmbedding: f"{MODEL_CACHE_DIR}/{self.model_name}", ) - def _load_model(self, path: str): + def _load_model(self, path: str) -> Optional[ort.InferenceSession]: if os.path.exists(path): + logger.debug( + f"loading ORT session with providers {self.providers} and options {self.provider_options}" + ) return ort.InferenceSession( path, providers=self.providers, provider_options=self.provider_options )