mirror of
https://github.com/blakeblackshear/frigate.git
synced 2026-03-17 13:48:21 +03:00
Merge 67dca94651 into ef5608a970
This commit is contained in:
commit
bfb817281c
@ -22,7 +22,7 @@ from .api import RealTimeProcessorApi
|
|||||||
try:
|
try:
|
||||||
from tflite_runtime.interpreter import Interpreter
|
from tflite_runtime.interpreter import Interpreter
|
||||||
except ModuleNotFoundError:
|
except ModuleNotFoundError:
|
||||||
from tensorflow.lite.python.interpreter import Interpreter
|
from ai_edge_litert.interpreter import Interpreter
|
||||||
|
|
||||||
logger = logging.getLogger(__name__)
|
logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
|
|||||||
@ -32,7 +32,7 @@ from .api import RealTimeProcessorApi
|
|||||||
try:
|
try:
|
||||||
from tflite_runtime.interpreter import Interpreter
|
from tflite_runtime.interpreter import Interpreter
|
||||||
except ModuleNotFoundError:
|
except ModuleNotFoundError:
|
||||||
from tensorflow.lite.python.interpreter import Interpreter
|
from ai_edge_litert.interpreter import Interpreter
|
||||||
|
|
||||||
logger = logging.getLogger(__name__)
|
logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
@ -76,7 +76,7 @@ class CustomStateClassificationProcessor(RealTimeProcessorApi):
|
|||||||
try:
|
try:
|
||||||
from tflite_runtime.interpreter import Interpreter
|
from tflite_runtime.interpreter import Interpreter
|
||||||
except ModuleNotFoundError:
|
except ModuleNotFoundError:
|
||||||
from tensorflow.lite.python.interpreter import Interpreter
|
from ai_edge_litert.interpreter import Interpreter
|
||||||
|
|
||||||
model_path = os.path.join(self.model_dir, "model.tflite")
|
model_path = os.path.join(self.model_dir, "model.tflite")
|
||||||
labelmap_path = os.path.join(self.model_dir, "labelmap.txt")
|
labelmap_path = os.path.join(self.model_dir, "labelmap.txt")
|
||||||
|
|||||||
@ -6,7 +6,7 @@ import numpy as np
|
|||||||
try:
|
try:
|
||||||
from tflite_runtime.interpreter import Interpreter, load_delegate
|
from tflite_runtime.interpreter import Interpreter, load_delegate
|
||||||
except ModuleNotFoundError:
|
except ModuleNotFoundError:
|
||||||
from tensorflow.lite.python.interpreter import Interpreter, load_delegate
|
from ai_edge_litert.interpreter import Interpreter, load_delegate
|
||||||
|
|
||||||
|
|
||||||
logger = logging.getLogger(__name__)
|
logger = logging.getLogger(__name__)
|
||||||
|
|||||||
@ -12,7 +12,7 @@ from ..detector_utils import tflite_detect_raw, tflite_init
|
|||||||
try:
|
try:
|
||||||
from tflite_runtime.interpreter import Interpreter
|
from tflite_runtime.interpreter import Interpreter
|
||||||
except ModuleNotFoundError:
|
except ModuleNotFoundError:
|
||||||
from tensorflow.lite.python.interpreter import Interpreter
|
from ai_edge_litert.interpreter import Interpreter
|
||||||
|
|
||||||
|
|
||||||
logger = logging.getLogger(__name__)
|
logger = logging.getLogger(__name__)
|
||||||
|
|||||||
@ -13,7 +13,7 @@ from frigate.detectors.detector_config import BaseDetectorConfig, ModelTypeEnum
|
|||||||
try:
|
try:
|
||||||
from tflite_runtime.interpreter import Interpreter, load_delegate
|
from tflite_runtime.interpreter import Interpreter, load_delegate
|
||||||
except ModuleNotFoundError:
|
except ModuleNotFoundError:
|
||||||
from tensorflow.lite.python.interpreter import Interpreter, load_delegate
|
from ai_edge_litert.interpreter import Interpreter, load_delegate
|
||||||
|
|
||||||
logger = logging.getLogger(__name__)
|
logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
|
|||||||
@ -17,7 +17,7 @@ from .base_embedding import BaseEmbedding
|
|||||||
try:
|
try:
|
||||||
from tflite_runtime.interpreter import Interpreter
|
from tflite_runtime.interpreter import Interpreter
|
||||||
except ModuleNotFoundError:
|
except ModuleNotFoundError:
|
||||||
from tensorflow.lite.python.interpreter import Interpreter
|
from ai_edge_litert.interpreter import Interpreter
|
||||||
|
|
||||||
logger = logging.getLogger(__name__)
|
logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
|
|||||||
@ -43,7 +43,7 @@ from frigate.video import start_or_restart_ffmpeg, stop_ffmpeg
|
|||||||
try:
|
try:
|
||||||
from tflite_runtime.interpreter import Interpreter
|
from tflite_runtime.interpreter import Interpreter
|
||||||
except ModuleNotFoundError:
|
except ModuleNotFoundError:
|
||||||
from tensorflow.lite.python.interpreter import Interpreter
|
from ai_edge_litert.interpreter import Interpreter
|
||||||
|
|
||||||
|
|
||||||
logger = logging.getLogger(__name__)
|
logger = logging.getLogger(__name__)
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user