Mirrors/blakeblackshear.frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2026-02-20 13:54:36 +01:00
Code Issues Projects Releases Wiki Activity

LPR device selection (#17592)

Browse Source 
* config changes

* use device

* docs

* docs

* reset state

* remove auto

* use device

* docs
This commit is contained in:
Josh Hawkins
2025-04-07 20:30:08 -05:00
committed by GitHub
parent cb27bdb2f7
commit 921908fe33
5 changed files with 27 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
frigate/data_processing/common/license_plate/model.py
View File

@@ -12,13 +12,13 @@ class LicensePlateModelRunner(DataProcessorModelRunner):
def __init__(self, requestor, device: str = "CPU", model_size: str = "large"):
super().__init__(requestor, device, model_size)
self.detection_model = PaddleOCRDetection(
model_size=model_size, requestor=requestor, device="CPU"
model_size=model_size, requestor=requestor, device=device
)
self.classification_model = PaddleOCRClassification(
model_size=model_size, requestor=requestor, device="CPU"
model_size=model_size, requestor=requestor, device=device
)
self.recognition_model = PaddleOCRRecognition(
model_size=model_size, requestor=requestor, device="CPU"
model_size=model_size, requestor=requestor, device=device
)
self.yolov9_detection_model = LicensePlateDetector(
model_size=model_size, requestor=requestor, device=device