* small tweaks for frigate+ submission and debug object list

* exclude attributes from labels colormap
This commit is contained in:
Josh Hawkins
2024-10-13 16:48:54 -05:00
committed by GitHub
parent 1ec459ea3a
commit 833768172d
3 changed files with 13 additions and 5 deletions

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)}