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

Add handling for non-existent or empty model file paths in compute_model_hash() method

Browse Source
This commit is contained in:
Sergey Krashevich 2023-04-26 04:11:28 +03:00
parent 17e8a46c7d
commit bebb22dc11
No known key found for this signature in database
GPG Key ID: 625171324E7D3856
1 changed files with 9 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
frigate/detectors/detector_config.py
View File

@ -118,11 +118,15 @@ class ModelConfig(BaseModel):
}
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):
print("File does not exist or path is empty.")
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."""