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

fix synaptics detector from throwing error when unused

Browse Source
This commit is contained in:
Josh Hawkins 2025-09-28 07:25:21 -05:00
parent f30934361b
commit 805170dd87
1 changed files with 15 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
frigate/detectors/plugins/synaptics.py
View File

@ -2,10 +2,6 @@ import logging
import os import os
import numpy as np import numpy as np
from synap import Network
from synap.postprocessor import Detector
from synap.preprocessor import Preprocessor
from synap.types import Layout, Shape
from typing_extensions import Literal from typing_extensions import Literal
from frigate.detectors.detection_api import DetectionApi from frigate.detectors.detection_api import DetectionApi
@ -15,6 +11,16 @@ from frigate.detectors.detector_config import (
ModelTypeEnum, ModelTypeEnum,
) )
try:
from synap import Network
from synap.postprocessor import Detector
from synap.preprocessor import Preprocessor
from synap.types import Layout, Shape
SYNAP_SUPPORT = True
except ImportError:
SYNAP_SUPPORT = False
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
DETECTOR_KEY = "synaptics" DETECTOR_KEY = "synaptics"
@ -28,6 +34,11 @@ class SynapDetector(DetectionApi):
type_key = DETECTOR_KEY type_key = DETECTOR_KEY
def __init__(self, detector_config: SynapDetectorConfig): def __init__(self, detector_config: SynapDetectorConfig):
if not SYNAP_SUPPORT:
raise ImportError(
"Error importing Synaptics SDK modules. You must use the -synaptics Docker image variant for Synaptics detector support."
)
try: try:
_, ext = os.path.splitext(detector_config.model.path) _, ext = os.path.splitext(detector_config.model.path)
if ext and ext != ".synap": if ext and ext != ".synap":