administrator/frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2026-02-04 02:05:21 +03:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

Don't fail if openvino fails to import

Browse Source
This commit is contained in:
Nicolas Mowen 2023-02-17 13:22:33 -07:00 committed by GitHub
parent c901707670
commit 18d43dcc5b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 15 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
frigate/detectors/plugins/openvino.py
View File

@ -1,15 +1,23 @@
import logging import logging
import numpy as np
import openvino.runtime as ov
from frigate.detectors.detection_api import DetectionApi
from frigate.detectors.detector_config import BaseDetectorConfig, ModelTypeEnum
from typing import Literal
from pydantic import Extra, Field
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
try:
# currently openvino may fail when importing
# on an arm device with 64 KiB page size
import openvino.runtime as ov
except ImportError as e:
logger.error(f"Error importing openvino runtime: {e}")
import numpy as np
from pydantic import Extra, Field
from typing import Literal
from frigate.detectors.detection_api import DetectionApi
from frigate.detectors.detector_config import BaseDetectorConfig, ModelTypeEnum
DETECTOR_KEY = "openvino" DETECTOR_KEY = "openvino"