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

Update attribute for attribute type object classification

Browse Source
This commit is contained in:
Nicolas Mowen 2025-07-18 07:50:16 -06:00
parent 69977e775d
commit 6f071d9fca
1 changed files with 8 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
frigate/data_processing/real_time/custom_classification.py
View File

@ -295,8 +295,14 @@ class CustomObjectClassificationProcessor(RealTimeProcessorApi):
EventMetadataTypeEnum.sub_label, EventMetadataTypeEnum.sub_label,
(obj_data["id"], sub_label, score), (obj_data["id"], sub_label, score),
) )
elif self.model_config.object_config.classification_type == ObjectClassificationType.attribute: elif (
pass 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): def handle_request(self, topic, request_data):
if topic == EmbeddingsRequestEnum.reload_classification_model.value: if topic == EmbeddingsRequestEnum.reload_classification_model.value: