forked from Github/frigate
Only set colors for enabled objects (#4936)
* Only create colormap for enabled labels * Fix assigning
This commit is contained in:
@@ -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:
|
||||
|
||||
Reference in New Issue
Block a user