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

Add 'device' key to semantic search and face recognition. Modify LPR device key.

Browse Source
This commit is contained in:
baudneo 2025-07-31 22:29:37 -06:00
parent 5a49d1f73c
commit cb03489a97
No known key found for this signature in database
GPG Key ID: 51445F2ED08EBC7F
1 changed files with 16 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
frigate/config/classification.py
View File

@ -130,6 +130,11 @@ class SemanticSearchConfig(FrigateBaseModel):
model_size: str = Field( model_size: str = Field(
default="small", title="The size of the embeddings model used." default="small", title="The size of the embeddings model used."
) )
device: str = Field(
default="CPU",
title="The device to use for license plate recognition.",
description="Use 'cpu' or 'gpu', to target a specific gpu use: '0', '1', etc.",
)
class TriggerConfig(FrigateBaseModel): class TriggerConfig(FrigateBaseModel):
@ -196,6 +201,11 @@ class FaceRecognitionConfig(FrigateBaseModel):
blur_confidence_filter: bool = Field( blur_confidence_filter: bool = Field(
default=True, title="Apply blur quality filter to face confidence." default=True, title="Apply blur quality filter to face confidence."
) )
device: str = Field(
default="CPU",
title="The device to use for license plate recognition.",
description="Use 'cpu' or 'gpu', to target a specific gpu use: '0', '1', etc.",
)
class CameraFaceRecognitionConfig(FrigateBaseModel): class CameraFaceRecognitionConfig(FrigateBaseModel):
@ -209,10 +219,6 @@ class CameraFaceRecognitionConfig(FrigateBaseModel):
class LicensePlateRecognitionConfig(FrigateBaseModel): class LicensePlateRecognitionConfig(FrigateBaseModel):
enabled: bool = Field(default=False, title="Enable license plate recognition.") enabled: bool = Field(default=False, title="Enable license plate recognition.")
device: Optional[EnrichmentsDeviceEnum] = Field(
default=EnrichmentsDeviceEnum.CPU,
title="The device used for license plate recognition.",
)
model_size: str = Field( model_size: str = Field(
default="small", title="The size of the embeddings model used." default="small", title="The size of the embeddings model used."
) )
@ -258,6 +264,11 @@ class LicensePlateRecognitionConfig(FrigateBaseModel):
default=False, default=False,
title="Save plates captured for LPR for debugging purposes.", title="Save plates captured for LPR for debugging purposes.",
) )
device: str = Field(
default="CPU",
title="The device to use for license plate recognition.",
description="Use 'cpu' or 'gpu', to target a specific gpu use: '0', '1', etc.",
)
class CameraLicensePlateRecognitionConfig(FrigateBaseModel): class CameraLicensePlateRecognitionConfig(FrigateBaseModel):
@ -278,4 +289,4 @@ class CameraLicensePlateRecognitionConfig(FrigateBaseModel):
le=10, le=10,
) )
model_config = ConfigDict(extra="forbid", protected_namespaces=()) model_config = ConfigDict(extra="forbid", protected_namespaces=())