Skip to content

Commit

Permalink
Update logging, starting support for deepstack
Browse files Browse the repository at this point in the history
  • Loading branch information
snowzach committed Feb 1, 2022
1 parent 6384e1c commit dd82383
Showing 1 changed file with 7 additions and 1 deletion.
8 changes: 7 additions & 1 deletion doods.py
Original file line number Diff line number Diff line change
Expand Up @@ -21,6 +21,12 @@
except ModuleNotFoundError:
logger.info('PyTorch not installed...')

try:
from detectors.deepstack import DeepStack
detectors['deepstack'] = DeepStack
except ModuleNotFoundError:
logger.info('DeepStack not installed...')

try:
from detectors.tensorflow2 import Tensorflow2
detectors['tensorflow2'] = Tensorflow2
Expand Down Expand Up @@ -59,7 +65,7 @@ def __init__(self, config):
try:
detector = detector_class(detector_config)
except Exception as e:
logger.error('Could not create detector: %s' % e)
logger.error('Could not create detector %s/%s: %s' % (detector_config.type, detector_config.name, e))
continue
logger.info('Registered detector type:%s name:%s', detector.config.type, detector.config.name)
self._detectors[detector_config.name] = detector
Expand Down

0 comments on commit dd82383

Please sign in to comment.