From 2a5675950f7c897e0fbb7f945d84347d05ac5ecf Mon Sep 17 00:00:00 2001 From: abinila siva <163017635+abinila4@users.noreply.github.com> Date: Fri, 25 Apr 2025 16:12:23 -0400 Subject: [PATCH] Minor cleanup --- frigate/detectors/plugins/memryx.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/frigate/detectors/plugins/memryx.py b/frigate/detectors/plugins/memryx.py index a326513b1..ec19a318d 100644 --- a/frigate/detectors/plugins/memryx.py +++ b/frigate/detectors/plugins/memryx.py @@ -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}")