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

Only support tflite with classification models

Browse Source
This commit is contained in:
Nicolas Mowen 2025-09-19 07:22:40 -06:00
parent 475cbe3a0f
commit 35925f21d7
1 changed files with 3 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
frigate/data_processing/real_time/custom_classification.py
View File

@ -7,6 +7,7 @@ from typing import Any
import cv2 import cv2
import numpy as np import numpy as np
from tflite_runtime.interpreter import Interpreter
from frigate.comms.embeddings_updater import EmbeddingsRequestEnum from frigate.comms.embeddings_updater import EmbeddingsRequestEnum
from frigate.comms.event_metadata_updater import ( from frigate.comms.event_metadata_updater import (
@ -39,12 +40,11 @@ class CustomStateClassificationProcessor(RealTimeProcessorApi):
metrics: DataProcessorMetrics, metrics: DataProcessorMetrics,
): ):
super().__init__(config, metrics) super().__init__(config, metrics)
self.model_config = model_config self.model_config = model_config
self.requestor = requestor self.requestor = requestor
self.model_dir = os.path.join(MODEL_CACHE_DIR, self.model_config.name) self.model_dir = os.path.join(MODEL_CACHE_DIR, self.model_config.name)
self.train_dir = os.path.join(CLIPS_DIR, self.model_config.name, "train") self.train_dir = os.path.join(CLIPS_DIR, self.model_config.name, "train")
self.interpreter = None self.interpreter: Interpreter = None
self.tensor_input_details: dict[str, Any] = None self.tensor_input_details: dict[str, Any] = None
self.tensor_output_details: dict[str, Any] = None self.tensor_output_details: dict[str, Any] = None
self.labelmap: dict[int, str] = {} self.labelmap: dict[int, str] = {}
@ -57,11 +57,6 @@ class CustomStateClassificationProcessor(RealTimeProcessorApi):
@redirect_output_to_logger(logger, logging.DEBUG) @redirect_output_to_logger(logger, logging.DEBUG)
def __build_detector(self) -> None: def __build_detector(self) -> None:
try:
from tflite_runtime.interpreter import Interpreter
except ModuleNotFoundError:
from tensorflow.lite.python.interpreter import Interpreter
self.interpreter = Interpreter( self.interpreter = Interpreter(
model_path=os.path.join(self.model_dir, "model.tflite"), model_path=os.path.join(self.model_dir, "model.tflite"),
num_threads=2, num_threads=2,
@ -198,7 +193,7 @@ class CustomObjectClassificationProcessor(RealTimeProcessorApi):
self.model_config = model_config self.model_config = model_config
self.model_dir = os.path.join(MODEL_CACHE_DIR, self.model_config.name) self.model_dir = os.path.join(MODEL_CACHE_DIR, self.model_config.name)
self.train_dir = os.path.join(CLIPS_DIR, self.model_config.name, "train") self.train_dir = os.path.join(CLIPS_DIR, self.model_config.name, "train")
self.interpreter = None self.interpreter: Interpreter = None
self.sub_label_publisher = sub_label_publisher self.sub_label_publisher = sub_label_publisher
self.tensor_input_details: dict[str, Any] = None self.tensor_input_details: dict[str, Any] = None
self.tensor_output_details: dict[str, Any] = None self.tensor_output_details: dict[str, Any] = None
@ -212,11 +207,6 @@ class CustomObjectClassificationProcessor(RealTimeProcessorApi):
@redirect_output_to_logger(logger, logging.DEBUG) @redirect_output_to_logger(logger, logging.DEBUG)
def __build_detector(self) -> None: def __build_detector(self) -> None:
try:
from tflite_runtime.interpreter import Interpreter
except ModuleNotFoundError:
from tensorflow.lite.python.interpreter import Interpreter
self.interpreter = Interpreter( self.interpreter = Interpreter(
model_path=os.path.join(self.model_dir, "model.tflite"), model_path=os.path.join(self.model_dir, "model.tflite"),
num_threads=2, num_threads=2,