administrator/frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2026-02-15 07:35:27 +03:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

Add debugging logs for more info

Browse Source
This commit is contained in:
Nicolas Mowen 2024-10-10 10:10:28 -06:00
parent f67ec241d4
commit 6f335800e9
2 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
frigate/embeddings/embeddings.py
View File

@ -168,7 +168,9 @@ class Embeddings:
logger.info("Indexing tracked object embeddings...") logger.info("Indexing tracked object embeddings...")
self._drop_tables() self._drop_tables()
logger.debug("Dropped embeddings tables.")
self._create_tables() self._create_tables()
logger.debug("Created embeddings tables.")
st = time.time() st = time.time()
totals = { totals = {

6
frigate/embeddings/functions/onnx.py
View File

@ -110,6 +110,7 @@ class GenericONNXEmbedding:
self.session = self._load_model( self.session = self._load_model(
os.path.join(self.download_path, self.model_file) os.path.join(self.download_path, self.model_file)
) )
logger.debug("successfully loaded model.")
def _load_tokenizer(self): def _load_tokenizer(self):
tokenizer_path = os.path.join(f"{MODEL_CACHE_DIR}/{self.model_name}/tokenizer") 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}", 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): if os.path.exists(path):
logger.debug(
f"loading ORT session with providers {self.providers} and options {self.provider_options}"
)
return ort.InferenceSession( return ort.InferenceSession(
path, providers=self.providers, provider_options=self.provider_options path, providers=self.providers, provider_options=self.provider_options
) )