Mirrors/blakeblackshear.frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2025-08-27 13:47:50 +02:00
Code Issues Projects Releases Wiki Activity

Minor cleanup

Browse Source
This commit is contained in:
abinila siva 2025-04-25 16:12:23 -04:00 committed by GitHub
parent 32bb3cd25d
commit 2a5675950f
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
frigate/detectors/plugins/memryx.py
View File

@ -173,7 +173,7 @@ class MemryXDetector(DetectionApi):
zip_path = os.path.join(self.cache_dir, f"{self.memx_model_type.value}.zip")
try:
# Before downloading, check if someone else already downloaded
# Before downloading, check if already downloaded
if not os.path.exists(zip_path):
# Download only if zip does not exist
urllib.request.urlretrieve(self.model_url, zip_path)
@ -197,8 +197,8 @@ class MemryXDetector(DetectionApi):
logger.info(f"Assigned Model Path: {self.memx_model_path}")
logger.info(f"Assigned Post-processing Model Path: {self.memx_post_model}")
if self.memx_model_type in [ModelTypeEnum.yolov8, ModelTypeEnum.yolov9]:
self.load_yolo_constants()
if self.memx_model_type == ModelTypeEnum.yolov9:
self.load_yolo_constants()
except Exception as e:
logger.error(f"Failed to prepare model: {e}")