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

Detection adjustments (#14329)

Browse Source
This commit is contained in:
Nicolas Mowen
2024-10-13 21:34:51 -05:00
committed by GitHub
parent 4ca267ea17
commit 9adffa1ef5
2 changed files with 21 additions and 28 deletions
Download Patch File Download Diff File Expand all files Collapse all files
frigate/detectors/detector_config.py
+6 -10
View File
@@ -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=())