From a8db97854185946d8da0df529c1a466de0781964 Mon Sep 17 00:00:00 2001 From: Josh Hawkins <32435876+hawkeye217@users.noreply.github.com> Date: Mon, 7 Apr 2025 14:52:59 -0500 Subject: [PATCH] always run paddleocr models on cpu --- frigate/data_processing/common/license_plate/model.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/frigate/data_processing/common/license_plate/model.py b/frigate/data_processing/common/license_plate/model.py index 25e7b2cafc..b22ab73a5f 100644 --- a/frigate/data_processing/common/license_plate/model.py +++ b/frigate/data_processing/common/license_plate/model.py @@ -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=device + model_size=model_size, requestor=requestor, device="CPU" ) self.classification_model = PaddleOCRClassification( - model_size=model_size, requestor=requestor, device=device + model_size=model_size, requestor=requestor, device="CPU" ) self.recognition_model = PaddleOCRRecognition( - model_size=model_size, requestor=requestor, device=device + model_size=model_size, requestor=requestor, device="CPU" ) self.yolov9_detection_model = LicensePlateDetector( model_size=model_size, requestor=requestor, device=device