From a4e125d54abd1ba0f27ab3433e9c6be98033fa4c Mon Sep 17 00:00:00 2001 From: Nicolas Mowen Date: Sun, 13 Oct 2024 19:38:19 -0600 Subject: [PATCH] Fix color map --- frigate/detectors/detector_config.py | 16 ++++++---------- 1 file changed, 6 insertions(+), 10 deletions(-) diff --git a/frigate/detectors/detector_config.py b/frigate/detectors/detector_config.py index 11f08a86c..90937d8f4 100644 --- a/frigate/detectors/detector_config.py +++ b/frigate/detectors/detector_config.py @@ -158,17 +158,13 @@ class ModelConfig(BaseModel): def create_colormap(self, enabled_labels: set[str]) -> None: """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 - ) - ) - ) + enabled_trackable_labels = list( + filter(lambda label: label not in self._all_attributes, enabled_labels) ) - - self._colormap = {label: color for label, color in zip(enabled_labels, colors)} + colors = generate_color_palette(len(enabled_trackable_labels)) + self._colormap = { + label: color for label, color in zip(enabled_trackable_labels, colors) + } model_config = ConfigDict(extra="forbid", protected_namespaces=())