administrator/frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2026-02-15 07:35:27 +03:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

exclude attributes from labels colormap

Browse Source
This commit is contained in:
Josh Hawkins 2024-10-13 16:39:14 -05:00
parent 96453f17f8
commit c7273580f4
1 changed files with 10 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
frigate/detectors/detector_config.py
View File

@ -157,8 +157,16 @@ class ModelConfig(BaseModel):
self._model_hash = file_hash.hexdigest()
def create_colormap(self, enabled_labels: set[str]) -> None:
"""Get a list of colors for enabled labels."""
colors = generate_color_palette(len(enabled_labels))
"""Get a list of colors for enabled labels that aren't attributes."""
colors = generate_color_palette(
len(
list(
filter(
lambda label: label not in self._all_attributes, enabled_labels
)
)
)
)
self._colormap = {label: color for label, color in zip(enabled_labels, colors)}