administrator/frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2026-02-03 17:55:21 +03:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

Fix assigning

Browse Source
This commit is contained in:
Nick Mowen 2023-01-06 13:13:55 -07:00
parent 2185456ea5
commit fc0ea57dfd
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
frigate/config.py
View File

@ -968,6 +968,8 @@ class FrigateConfig(FrigateBaseModel):
for _, camera in config.cameras.items(): for _, camera in config.cameras.items():
enabled_labels.update(camera.objects.track) enabled_labels.update(camera.objects.track)
config.model.create_colormap(enabled_labels)
for key, detector in config.detectors.items(): for key, detector in config.detectors.items():
detector_config: DetectorConfig = parse_obj_as(DetectorConfig, detector) detector_config: DetectorConfig = parse_obj_as(DetectorConfig, detector)
if detector_config.model is None: if detector_config.model is None:
@ -992,7 +994,6 @@ class FrigateConfig(FrigateBaseModel):
config.model.dict(exclude_unset=True), config.model.dict(exclude_unset=True),
) )
detector_config.model = ModelConfig.parse_obj(merged_model) detector_config.model = ModelConfig.parse_obj(merged_model)
detector_config.model.create_colormap(enabled_labels)
config.detectors[key] = detector_config config.detectors[key] = detector_config
return config return config