Add handling for non-existent or empty model file paths (blakeblacksh…

…ear#6525) * Add handling for non-existent or empty model file paths in `compute_model_hash()` method * Remove print() in detector_config.py Co-authored-by: Blake Blackshear <[email protected]> --------- Co-authored-by: Blake Blackshear <[email protected]>
gilankpam · May 21, 2023 · 1173811 · 1173811
1 parent 53d63e0
commit 1173811
Showing 1 changed file with 8 additions and 5 deletions.
diff --git a/frigate/detectors/detector_config.py b/frigate/detectors/detector_config.py
@@ -118,11 +118,14 @@ def check_and_load_plus_model(
         }
 
     def compute_model_hash(self) -> None:
-        with open(self.path, "rb") as f:
-            file_hash = hashlib.md5()
-            while chunk := f.read(8192):
-                file_hash.update(chunk)
-        self._model_hash = file_hash.hexdigest()
+        if not self.path or not os.path.exists(self.path):
+            self._model_hash = hashlib.md5(b"unknown").hexdigest()
+        else:
+            with open(self.path, "rb") as f:
+                file_hash = hashlib.md5()
+                while chunk := f.read(8192):
+                    file_hash.update(chunk)
+            self._model_hash = file_hash.hexdigest()
 
     def create_colormap(self, enabled_labels: set[str]) -> None:
         """Get a list of colors for enabled labels."""