administrator/frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2026-02-18 09:04:28 +03:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

Initial setup for bird processing

Browse Source
This commit is contained in:
Nicolas Mowen 2025-01-11 09:11:55 -07:00
parent 7d478be798
commit 75c9a723bf
2 changed files with 8 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
frigate/data_processing/real_time/bird_processor.py
View File

@ -8,8 +8,8 @@ import numpy as np
from frigate.config import FrigateConfig
from frigate.const import MODEL_CACHE_DIR
from .processor_api import ProcessorApi
from .types import PostProcessingMetrics
from ..types import DataProcessorMetrics
from .api import RealTimeProcessorApi
try:
from tflite_runtime.interpreter import Interpreter
@ -19,8 +19,8 @@ except ModuleNotFoundError:
logger = logging.getLogger(__name__)
class BirdProcessor(ProcessorApi):
def __init__(self, config: FrigateConfig, metrics: PostProcessingMetrics):
class BirdProcessor(RealTimeProcessorApi):
def __init__(self, config: FrigateConfig, metrics: DataProcessorMetrics):
super().__init__(config, metrics)
self.interpreter: Interpreter = None
self.tensor_input_details: dict[str, any] = None

3
frigate/embeddings/maintainer.py
View File

@ -30,6 +30,7 @@ from frigate.const import (
UPDATE_EVENT_DESCRIPTION,
)
from frigate.data_processing.real_time.api import RealTimeProcessorApi
from frigate.data_processing.real_time.bird_processor import BirdProcessor
from frigate.data_processing.real_time.face_processor import FaceProcessor
from frigate.data_processing.types import DataProcessorMetrics
from frigate.embeddings.lpr.lpr import LicensePlateRecognition
@ -78,6 +79,8 @@ class EmbeddingMaintainer(threading.Thread):
if self.config.face_recognition.enabled:
self.processors.append(FaceProcessor(self.config, metrics))
self.processors.append(BirdProcessor(self.config, metrics))
# create communication for updating event descriptions
self.requestor = InterProcessRequestor()
self.stop_event = stop_event