Update recognized license plate topic to be used as attribute updater

frigate/api/event.py

@@ -1148,7 +1148,8 @@ def set_plate(
         new_score = None
 
     request.app.event_metadata_updater.publish(
-        EventMetadataTypeEnum.recognized_license_plate, (event_id, new_plate, new_score)
+        EventMetadataTypeEnum.attribute,
+        (event_id, "recognized_license_plate", new_plate, new_score),
     )
 
     return JSONResponse(

frigate/comms/event_metadata_updater.py

@@ -15,7 +15,7 @@ class EventMetadataTypeEnum(str, Enum):
     manual_event_end = "manual_event_end"
     regenerate_description = "regenerate_description"
     sub_label = "sub_label"
-    recognized_license_plate = "recognized_license_plate"
+    attribute = "attribute"
     lpr_event_create = "lpr_event_create"
     save_lpr_snapshot = "save_lpr_snapshot"
 

frigate/data_processing/common/license_plate/mixin.py

@@ -1537,8 +1537,8 @@ class LicensePlateProcessingMixin:
             ),
         )
         self.sub_label_publisher.publish(
-            EventMetadataTypeEnum.recognized_license_plate,
-            (id, top_plate, avg_confidence),
+            EventMetadataTypeEnum.attribute,
+            (id, "recognized_license_plate", top_plate, avg_confidence),
         )
 
         # save the best snapshot for dedicated lpr cams not using frigate+

frigate/data_processing/real_time/custom_classification.py

@@ -18,6 +18,7 @@ from frigate.config import FrigateConfig
 from frigate.config.classification import CustomClassificationConfig
 from frigate.const import CLIPS_DIR, MODEL_CACHE_DIR
 from frigate.log import redirect_output_to_logger
+from frigate.types import ObjectClassificationType
 from frigate.util.builtin import EventsPerSecond, InferenceSpeed, load_labels
 from frigate.util.object import box_overlaps, calculate_region
 
@@ -285,11 +286,17 @@ class CustomObjectClassificationProcessor(RealTimeProcessorApi):
         sub_label = self.labelmap[best_id]
         self.detected_objects[obj_data["id"]] = score
 
-        if sub_label != "none":
-            self.sub_label_publisher.publish(
-                EventMetadataTypeEnum.sub_label,
-                (obj_data["id"], sub_label, score),
-            )
+        if (
+            self.model_config.object_config.classification_type
+            == ObjectClassificationType.sub_label
+        ):
+            if sub_label != "none":
+                self.sub_label_publisher.publish(
+                    EventMetadataTypeEnum.sub_label,
+                    (obj_data["id"], sub_label, score),
+                )
+        elif self.model_config.object_config.classification_type == ObjectClassificationType.attribute:
+            pass
 
     def handle_request(self, topic, request_data):
         if topic == EmbeddingsRequestEnum.reload_classification_model.value:

frigate/track/object_processing.py

@@ -376,10 +376,14 @@ class TrackedObjectProcessor(threading.Thread):
 
         return True
 
-    def set_recognized_license_plate(
-        self, event_id: str, recognized_license_plate: str | None, score: float | None
+    def set_object_attribute(
+        self,
+        event_id: str,
+        field_name: str,
+        field_value: str | None,
+        score: float | None,
     ) -> None:
-        """Update recognized license plate for given event id."""
+        """Update attribute for given event id."""
         tracked_obj: TrackedObject = None
 
         for state in self.camera_states.values():
@@ -397,18 +401,18 @@ class TrackedObjectProcessor(threading.Thread):
             return
 
         if tracked_obj:
-            tracked_obj.obj_data["recognized_license_plate"] = (
-                recognized_license_plate,
+            tracked_obj.obj_data[field_name] = (
+                field_value,
                 score,
             )
 
         if event:
             data = event.data
-            data["recognized_license_plate"] = recognized_license_plate
-            if recognized_license_plate is None:
-                data["recognized_license_plate_score"] = None
+            data[field_name] = field_value
+            if field_value is None:
+                data[f"{field_name}_score"] = None
             elif score is not None:
-                data["recognized_license_plate_score"] = score
+                data[f"{field_name}_score"] = score
             event.data = data
             event.save()
 
@@ -644,11 +648,9 @@ class TrackedObjectProcessor(threading.Thread):
                 if topic.endswith(EventMetadataTypeEnum.sub_label.value):
                     (event_id, sub_label, score) = payload
                     self.set_sub_label(event_id, sub_label, score)
-                if topic.endswith(EventMetadataTypeEnum.recognized_license_plate.value):
-                    (event_id, recognized_license_plate, score) = payload
-                    self.set_recognized_license_plate(
-                        event_id, recognized_license_plate, score
-                    )
+                if topic.endswith(EventMetadataTypeEnum.attribute.value):
+                    (event_id, field_name, field_value, score) = payload
+                    self.set_object_attribute(event_id, field_name, field_value, score)
                 elif topic.endswith(EventMetadataTypeEnum.lpr_event_create.value):
                     self.create_lpr_event(payload)
                 elif topic.endswith(EventMetadataTypeEnum.save_lpr_snapshot.value):