Only set colors for enabled objects (#4936)

* Only create colormap for enabled labels

* Fix assigning
This commit is contained in:
Nicolas Mowen
2023-01-06 18:31:54 -07:00
committed by GitHub
parent d9c45a76fe
commit 543cad5497
2 changed files with 14 additions and 4 deletions

View File

@@ -962,6 +962,14 @@ class FrigateConfig(FrigateBaseModel):
camera_config.create_ffmpeg_cmds()
config.cameras[name] = camera_config
# get list of unique enabled labels for tracking
enabled_labels = set(config.objects.track)
for _, camera in config.cameras.items():
enabled_labels.update(camera.objects.track)
config.model.create_colormap(enabled_labels)
for key, detector in config.detectors.items():
detector_config: DetectorConfig = parse_obj_as(DetectorConfig, detector)
if detector_config.model is None: