administrator/frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2026-07-17 09:21:13 +03:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

conditionally import axengine module

Browse Source 
to match other detectors
This commit is contained in:
Josh Hawkins 2026-03-20 07:59:58 -05:00
parent 46c96fbc5e
commit 43bfd0c0a1
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
frigate/detectors/plugins/axengine.py
View File

@ -4,7 +4,6 @@ import re
import urllib.request import urllib.request
from typing import Literal from typing import Literal
import axengine as axe
from pydantic import ConfigDict from pydantic import ConfigDict
from frigate.const import MODEL_CACHE_DIR from frigate.const import MODEL_CACHE_DIR
@ -37,6 +36,12 @@ class Axengine(DetectionApi):
type_key = DETECTOR_KEY type_key = DETECTOR_KEY
def __init__(self, config: AxengineDetectorConfig): def __init__(self, config: AxengineDetectorConfig):
try:
import axengine as axe
except ModuleNotFoundError:
raise ImportError("AXEngine is not installed.")
return
logger.info("__init__ axengine") logger.info("__init__ axengine")
super().__init__(config) super().__init__(config)
self.height = config.model.height self.height = config.model.height