From 6f071d9fca462dbc637d36fcd4314ef0b125966d Mon Sep 17 00:00:00 2001 From: Nicolas Mowen Date: Fri, 18 Jul 2025 07:50:16 -0600 Subject: [PATCH] Update attribute for attribute type object classification --- .../data_processing/real_time/custom_classification.py | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) diff --git a/frigate/data_processing/real_time/custom_classification.py b/frigate/data_processing/real_time/custom_classification.py index ac2081fad..434e8b500 100644 --- a/frigate/data_processing/real_time/custom_classification.py +++ b/frigate/data_processing/real_time/custom_classification.py @@ -295,8 +295,14 @@ class CustomObjectClassificationProcessor(RealTimeProcessorApi): EventMetadataTypeEnum.sub_label, (obj_data["id"], sub_label, score), ) - elif self.model_config.object_config.classification_type == ObjectClassificationType.attribute: - pass + elif ( + self.model_config.object_config.classification_type + == ObjectClassificationType.attribute + ): + self.sub_label_publisher.publish( + EventMetadataTypeEnum.attribute, + (obj_data["id"], self.model_config.name, sub_label, score), + ) def handle_request(self, topic, request_data): if topic == EmbeddingsRequestEnum.reload_classification_model.value: