From dd82383f1a10a1b2866492e77d229c3d24575364 Mon Sep 17 00:00:00 2001 From: Zach Brown Date: Mon, 31 Jan 2022 22:57:50 -0500 Subject: [PATCH] Update logging, starting support for deepstack --- doods.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/doods.py b/doods.py index 7310951..142f3a2 100644 --- a/doods.py +++ b/doods.py @@ -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 @@ -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