Mirrors/blakeblackshear.frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2025-09-05 17:51:36 +02:00
Code Issues Projects Releases Wiki Activity

Only import degirum module if using degirum detector (#19802)

Browse Source
This commit is contained in:
Josh Hawkins 2025-08-27 09:31:01 -05:00 committed by GitHub
parent 0febc4d456
commit d78b6e528b
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
frigate/detectors/plugins/degirum.py
View File

@ -1,7 +1,6 @@
import logging import logging
import queue import queue
import degirum as dg
import numpy as np import numpy as np
from pydantic import Field from pydantic import Field
from typing_extensions import Literal from typing_extensions import Literal
@ -26,6 +25,11 @@ class DGDetector(DetectionApi):
type_key = DETECTOR_KEY type_key = DETECTOR_KEY
def __init__(self, detector_config: DGDetectorConfig): def __init__(self, detector_config: DGDetectorConfig):
try:
import degirum as dg
except ModuleNotFoundError:
raise ImportError("Unable to import DeGirum detector.")
self._queue = queue.Queue() self._queue = queue.Queue()
self._zoo = dg.connect( self._zoo = dg.connect(
detector_config.location, detector_config.zoo, detector_config.token detector_config.location, detector_config.zoo, detector_config.token