mirror of
https://github.com/blakeblackshear/frigate.git
synced 2026-04-05 14:47:40 +03:00
Fix mixin types missing
This commit is contained in:
parent
206261c322
commit
9a654e9c4a
@ -22,19 +22,35 @@ from frigate.comms.event_metadata_updater import (
|
|||||||
EventMetadataPublisher,
|
EventMetadataPublisher,
|
||||||
EventMetadataTypeEnum,
|
EventMetadataTypeEnum,
|
||||||
)
|
)
|
||||||
|
from frigate.comms.inter_process import InterProcessRequestor
|
||||||
|
from frigate.config import FrigateConfig
|
||||||
|
from frigate.config.classification import LicensePlateRecognitionConfig
|
||||||
from frigate.const import CLIPS_DIR, MODEL_CACHE_DIR
|
from frigate.const import CLIPS_DIR, MODEL_CACHE_DIR
|
||||||
|
from frigate.data_processing.common.license_plate.model import LicensePlateModelRunner
|
||||||
from frigate.embeddings.onnx.lpr_embedding import LPR_EMBEDDING_SIZE
|
from frigate.embeddings.onnx.lpr_embedding import LPR_EMBEDDING_SIZE
|
||||||
from frigate.types import TrackedObjectUpdateTypesEnum
|
from frigate.types import TrackedObjectUpdateTypesEnum
|
||||||
from frigate.util.builtin import EventsPerSecond, InferenceSpeed
|
from frigate.util.builtin import EventsPerSecond, InferenceSpeed
|
||||||
from frigate.util.image import area
|
from frigate.util.image import area
|
||||||
|
|
||||||
|
from ...types import DataProcessorMetrics
|
||||||
|
|
||||||
logger = logging.getLogger(__name__)
|
logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
WRITE_DEBUG_IMAGES = False
|
WRITE_DEBUG_IMAGES = False
|
||||||
|
|
||||||
|
|
||||||
class LicensePlateProcessingMixin:
|
class LicensePlateProcessingMixin:
|
||||||
def __init__(self, *args, **kwargs):
|
# Attributes expected from consuming classes (set before super().__init__)
|
||||||
|
config: FrigateConfig
|
||||||
|
metrics: DataProcessorMetrics
|
||||||
|
model_runner: LicensePlateModelRunner
|
||||||
|
lpr_config: LicensePlateRecognitionConfig
|
||||||
|
requestor: InterProcessRequestor
|
||||||
|
detected_license_plates: dict[str, dict[str, Any]]
|
||||||
|
camera_current_cars: dict[str, list[str]]
|
||||||
|
sub_label_publisher: EventMetadataPublisher
|
||||||
|
|
||||||
|
def __init__(self, *args: Any, **kwargs: Any) -> None:
|
||||||
super().__init__(*args, **kwargs)
|
super().__init__(*args, **kwargs)
|
||||||
self.plate_rec_speed = InferenceSpeed(self.metrics.alpr_speed)
|
self.plate_rec_speed = InferenceSpeed(self.metrics.alpr_speed)
|
||||||
self.plates_rec_second = EventsPerSecond()
|
self.plates_rec_second = EventsPerSecond()
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user