mirror of
https://github.com/blakeblackshear/frigate.git
synced 2026-05-05 04:57:42 +03:00
default to small model
This commit is contained in:
parent
9476ca0dc3
commit
78bf8880f9
@ -9,7 +9,7 @@ from ...types import DataProcessorModelRunner
|
|||||||
|
|
||||||
|
|
||||||
class LicensePlateModelRunner(DataProcessorModelRunner):
|
class LicensePlateModelRunner(DataProcessorModelRunner):
|
||||||
def __init__(self, requestor, device: str = "CPU", model_size: str = "large"):
|
def __init__(self, requestor, device: str = "CPU", model_size: str = "small"):
|
||||||
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=device
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user