mirror of
https://github.com/blakeblackshear/frigate.git
synced 2026-02-18 09:04:28 +03:00
Refactor post processor to be real time processor
This commit is contained in:
parent
b51ac37496
commit
a1cc2449a0
@ -5,14 +5,14 @@ import numpy as np
|
||||
|
||||
from frigate.config import FrigateConfig
|
||||
|
||||
from .types import PostProcessingMetrics
|
||||
from ..types import DataProcessorMetrics
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
|
||||
class ProcessorApi(ABC):
|
||||
class RealTimeProcessorApi(ABC):
|
||||
@abstractmethod
|
||||
def __init__(self, config: FrigateConfig, metrics: PostProcessingMetrics) -> None:
|
||||
def __init__(self, config: FrigateConfig, metrics: DataProcessorMetrics) -> None:
|
||||
self.config = config
|
||||
self.metrics = metrics
|
||||
pass
|
||||
@ -16,8 +16,8 @@ from frigate.config import FrigateConfig
|
||||
from frigate.const import FACE_DIR, FRIGATE_LOCALHOST, MODEL_CACHE_DIR
|
||||
from frigate.util.image import area
|
||||
|
||||
from .processor_api import ProcessorApi
|
||||
from .types import PostProcessingMetrics
|
||||
from ..types import DataProcessorMetrics
|
||||
from .api import RealTimeProcessorApi
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
@ -25,8 +25,8 @@ logger = logging.getLogger(__name__)
|
||||
MIN_MATCHING_FACES = 2
|
||||
|
||||
|
||||
class FaceProcessor(ProcessorApi):
|
||||
def __init__(self, config: FrigateConfig, metrics: PostProcessingMetrics):
|
||||
class FaceProcessor(RealTimeProcessorApi):
|
||||
def __init__(self, config: FrigateConfig, metrics: DataProcessorMetrics):
|
||||
super().__init__(config, metrics)
|
||||
self.face_config = config.face_recognition
|
||||
self.face_detector: cv2.FaceDetectorYN = None
|
||||
@ -4,7 +4,7 @@ import multiprocessing as mp
|
||||
from multiprocessing.sharedctypes import Synchronized
|
||||
|
||||
|
||||
class PostProcessingMetrics:
|
||||
class DataProcessorMetrics:
|
||||
image_embeddings_fps: Synchronized
|
||||
text_embeddings_sps: Synchronized
|
||||
face_rec_fps: Synchronized
|
||||
Loading…
Reference in New Issue
Block a user