administrator/frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2026-05-09 15:05:26 +03:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

Fix #3

Browse Source
This commit is contained in:
OmriAx 2025-03-10 12:28:04 +02:00
parent c674d142dc
commit 2811886498
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
frigate/detectors/plugins/hailo8l.py
View File

@ -343,7 +343,7 @@ class HailoDetector(DetectionApi):
raise ValueError("Invalid model URL. Only .hef files are supported.") raise ValueError("Invalid model URL. Only .hef files are supported.")
try: try:
urllib.request.urlretrieve(url, destination) urllib.request.urlretrieve(url, destination)
print(f"Downloaded model to {destination}") logger.debug(f"Downloaded model to {destination}")
except Exception as e: except Exception as e:
raise RuntimeError(f"Failed to download model from {url}: {str(e)}") raise RuntimeError(f"Failed to download model from {url}: {str(e)}")
@ -354,20 +354,20 @@ class HailoDetector(DetectionApi):
cached_model_path = os.path.join(self.cache_dir, model_name) cached_model_path = os.path.join(self.cache_dir, model_name)
if not self.model_path and not self.url: if not self.model_path and not self.url:
if os.path.exists(cached_model_path): if os.path.exists(cached_model_path):
print(f"Model found in cache: {cached_model_path}") logger.debug(f"Model found in cache: {cached_model_path}")
return cached_model_path return cached_model_path
else: else:
print(f"Downloading default model: {model_name}") logger.debug(f"Downloading default model: {model_name}")
if ARCH == "hailo8": if ARCH == "hailo8":
self.download_model(H8_DEFAULT_URL, cached_model_path) self.download_model(H8_DEFAULT_URL, cached_model_path)
else: else:
self.download_model(H8L_DEFAULT_URL, cached_model_path) self.download_model(H8L_DEFAULT_URL, cached_model_path)
elif self.url: elif self.url:
print(f"Downloading model from URL: {self.url}") logger.debug(f"Downloading model from URL: {self.url}")
self.download_model(self.url, cached_model_path) self.download_model(self.url, cached_model_path)
elif self.model_path: elif self.model_path:
if os.path.exists(self.model_path): if os.path.exists(self.model_path):
print(f"Using existing model at: {self.model_path}") logger.debug(f"Using existing model at: {self.model_path}")
return self.model_path return self.model_path
else: else:
raise FileNotFoundError(f"Model file not found at: {self.model_path}") raise FileNotFoundError(f"Model file not found at: {self.model_path}")