administrator/frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2026-05-05 13:07:44 +03:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

Add to type enum

Browse Source
This commit is contained in:
Nicolas Mowen 2025-04-28 11:46:36 -06:00
parent 9b840c3def
commit e15924606b
3 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
frigate/data_processing/common/license_plate/mixin.py
View File

@ -24,6 +24,7 @@ from frigate.comms.event_metadata_updater import (
) )
from frigate.const import CLIPS_DIR from frigate.const import CLIPS_DIR
from frigate.embeddings.onnx.lpr_embedding import LPR_EMBEDDING_SIZE from frigate.embeddings.onnx.lpr_embedding import LPR_EMBEDDING_SIZE
from frigate.types import TrackedObjectUpdateTypesEnum
from frigate.util.builtin import EventsPerSecond from frigate.util.builtin import EventsPerSecond
from frigate.util.image import area from frigate.util.image import area
@ -1514,7 +1515,7 @@ class LicensePlateProcessingMixin:
self.requestor.send_data( self.requestor.send_data(
"tracked_object_update", "tracked_object_update",
json.dumps({ json.dumps({
"type": "lpr", "type": TrackedObjectUpdateTypesEnum.lpr,
"name": sub_label, "name": sub_label,
"plate": top_plate, "plate": top_plate,
"score": avg_confidence, "score": avg_confidence,

3
frigate/data_processing/real_time/face.py
View File

@ -26,6 +26,7 @@ from frigate.data_processing.common.face.model import (
FaceNetRecognizer, FaceNetRecognizer,
FaceRecognizer, FaceRecognizer,
) )
from frigate.types import TrackedObjectUpdateTypesEnum
from frigate.util.builtin import EventsPerSecond from frigate.util.builtin import EventsPerSecond
from frigate.util.image import area from frigate.util.image import area
@ -295,7 +296,7 @@ class FaceRealTimeProcessor(RealTimeProcessorApi):
self.requestor.send_data( self.requestor.send_data(
"tracked_object_update", "tracked_object_update",
json.dumps({ json.dumps({
"type": "face", "type": TrackedObjectUpdateTypesEnum.face,
"name": weighted_sub_label, "name": weighted_sub_label,
"score": weighted_score, "score": weighted_score,
"id": id, "id": id,

2
frigate/types.py
View File

@ -25,3 +25,5 @@ class ModelStatusTypesEnum(str, Enum):
class TrackedObjectUpdateTypesEnum(str, Enum): class TrackedObjectUpdateTypesEnum(str, Enum):
description = "description" description = "description"
face = "face"
lpr = "lpr"