Add handling for non-existent or empty model file paths (#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]>
blakeblackshear · 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."""