Mirrors/blakeblackshear.frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2025-06-09 01:16:08 +02:00
Code Issues Projects Releases Wiki Activity

Don't fail if openvino fails to import (#5532)

Browse Source 
* Don't fail if openvino fails to import

* Ensure all modules are imported safely

* Undo

* Fix list append
This commit is contained in:
Nicolas Mowen 2023-02-19 06:41:14 -07:00 committed by GitHub
parent 0592c8b0e2
commit 2797a60d4f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 13 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
frigate/detectors/detector_types.py
View File

@ -13,12 +13,19 @@ from .detector_config import BaseDetectorConfig
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
plugin_modules = [
importlib.import_module(name) _included_modules = pkgutil.iter_modules(plugins.__path__, plugins.__name__ + ".")
for finder, name, ispkg in pkgutil.iter_modules(
plugins.__path__, plugins.__name__ + "." plugin_modules = []
)
] for _, name, _ in _included_modules:
try:
# currently openvino may fail when importing
# on an arm device with 64 KiB page size.
plugin_modules.append(importlib.import_module(name))
except ImportError as e:
logger.error(f"Error importing detector runtime: {e}")
api_types = {det.type_key: det for det in DetectionApi.__subclasses__()} api_types = {det.type_key: det for det in DetectionApi.__subclasses__()}