Mirrors/blakeblackshear.frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2025-06-09 01:16:08 +02:00
Code Issues Projects Releases Wiki Activity

Fix case where model path is set but not labelmap path (#10215)

Browse Source
This commit is contained in:
Nicolas Mowen 2024-03-03 17:17:56 -07:00 committed by GitHub
parent 1441a83151
commit fa0f509e18
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
frigate/util/builtin.py
View File

@ -9,7 +9,7 @@ import urllib.parse
from collections import Counter from collections import Counter
from collections.abc import Mapping from collections.abc import Mapping
from pathlib import Path from pathlib import Path
from typing import Any, Tuple from typing import Any, Optional, Tuple
import numpy as np import numpy as np
import pytz import pytz
@ -139,7 +139,7 @@ def get_ffmpeg_arg_list(arg: Any) -> list:
return arg if isinstance(arg, list) else shlex.split(arg) return arg if isinstance(arg, list) else shlex.split(arg)
def load_labels(path, encoding="utf-8", prefill=91): def load_labels(path: Optional[str], encoding="utf-8", prefill=91):
"""Loads labels from file (with or without index numbers). """Loads labels from file (with or without index numbers).
Args: Args:
path: path to label file. path: path to label file.
@ -147,6 +147,9 @@ def load_labels(path, encoding="utf-8", prefill=91):
Returns: Returns:
Dictionary mapping indices to labels. Dictionary mapping indices to labels.
""" """
if path is None:
return {}
with open(path, "r", encoding=encoding) as f: with open(path, "r", encoding=encoding) as f:
labels = {index: "unknown" for index in range(prefill)} labels = {index: "unknown" for index in range(prefill)}
lines = f.readlines() lines = f.readlines()