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

always run paddleocr models on cpu

Browse Source
This commit is contained in:
Josh Hawkins 2025-04-07 14:52:59 -05:00
parent 6b6f4477da
commit a8db978541
1 changed files with 3 additions and 3 deletions
Show Stats 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"): def __init__(self, requestor, device: str = "CPU", model_size: str = "large"):
super().__init__(requestor, device, model_size) super().__init__(requestor, device, model_size)
self.detection_model = PaddleOCRDetection( self.detection_model = PaddleOCRDetection(
model_size=model_size, requestor=requestor, device=device model_size=model_size, requestor=requestor, device="CPU"
) )
self.classification_model = PaddleOCRClassification( self.classification_model = PaddleOCRClassification(
model_size=model_size, requestor=requestor, device=device model_size=model_size, requestor=requestor, device="CPU"
) )
self.recognition_model = PaddleOCRRecognition( 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( self.yolov9_detection_model = LicensePlateDetector(
model_size=model_size, requestor=requestor, device=device model_size=model_size, requestor=requestor, device=device