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

Update openvino

Browse Source
This commit is contained in:
Nicolas Mowen 2024-09-20 13:43:24 -06:00
parent 535bf6e4b9
commit a23334faa2
4 changed files with 6 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
docker/main/requirements-wheels-post.txt
View File

@ -1,3 +1,3 @@
# ONNX # ONNX
onnxruntime-openvino == 1.18.* ; platform_machine == 'x86_64' onnxruntime-openvino == 1.19.* ; platform_machine == 'x86_64'
onnxruntime == 1.18.* ; platform_machine == 'aarch64' onnxruntime == 1.19.* ; platform_machine == 'aarch64'

4
docker/main/requirements-wheels.txt
View File

@ -28,8 +28,8 @@ norfair == 2.2.*
setproctitle == 1.3.* setproctitle == 1.3.*
ws4py == 0.5.* ws4py == 0.5.*
unidecode == 1.3.* unidecode == 1.3.*
# OpenVino & ONNX # OpenVino (ONNX installed in wheels-post)
openvino == 2024.1.* openvino == 2024.3.*
# Embeddings # Embeddings
chromadb == 0.5.0 chromadb == 0.5.0
onnx_clip == 4.0.* onnx_clip == 4.0.*

3
frigate/detectors/plugins/onnx.py
View File

@ -17,6 +17,7 @@ DETECTOR_KEY = "onnx"
class ONNXDetectorConfig(BaseDetectorConfig): class ONNXDetectorConfig(BaseDetectorConfig):
type: Literal[DETECTOR_KEY] type: Literal[DETECTOR_KEY]
device: str = Field(default=None, title="Device Type")
class ONNXDetector(DetectionApi): class ONNXDetector(DetectionApi):
@ -57,7 +58,7 @@ class ONNXDetector(DetectionApi):
options.append( options.append(
{ {
"cache_dir": "/config/model_cache/openvino/ort", "cache_dir": "/config/model_cache/openvino/ort",
"device_type": "GPU", "device_type": "AUTO",
} }
) )
else: else:

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

@ -30,12 +30,6 @@ class OvDetector(DetectionApi):
self.h = detector_config.model.height self.h = detector_config.model.height
self.w = detector_config.model.width self.w = detector_config.model.width
if detector_config.device == "AUTO":
logger.warning(
"OpenVINO AUTO device type is not currently supported. Attempting to use GPU instead."
)
detector_config.device = "GPU"
if not os.path.isfile(detector_config.model.path): if not os.path.isfile(detector_config.model.path):
logger.error(f"OpenVino model file {detector_config.model.path} not found.") logger.error(f"OpenVino model file {detector_config.model.path} not found.")
raise FileNotFoundError raise FileNotFoundError