mirror of
https://github.com/blakeblackshear/frigate.git
synced 2025-08-31 13:48:19 +02:00
Only import degirum module if using degirum detector (#19802)
This commit is contained in:
parent
0febc4d456
commit
d78b6e528b
@ -1,7 +1,6 @@
|
||||
import logging
|
||||
import queue
|
||||
|
||||
import degirum as dg
|
||||
import numpy as np
|
||||
from pydantic import Field
|
||||
from typing_extensions import Literal
|
||||
@ -26,6 +25,11 @@ class DGDetector(DetectionApi):
|
||||
type_key = DETECTOR_KEY
|
||||
|
||||
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._zoo = dg.connect(
|
||||
detector_config.location, detector_config.zoo, detector_config.token
|
||||
|
Loading…
Reference in New Issue
Block a user