Mirrors/blakeblackshear.frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2024-11-21 19:07:46 +01:00
Code Issues Projects Releases Wiki Activity

OpenVino - Error clarification for compile_model function (#12304)

Browse Source 
* Error clarification for openvino's compile_model function

* run ruff format

---------

Co-authored-by: ubawurinna <you@example.com>
This commit is contained in:
ubawurinna 2024-07-08 15:13:33 +02:00 committed by GitHub
parent a0741aa7b1
commit 2ea1d34f4f
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -1,4 +1,5 @@
import logging import logging
import os
import numpy as np import numpy as np
import openvino as ov import openvino as ov
@ -35,6 +36,10 @@ class OvDetector(DetectionApi):
) )
detector_config.device = "GPU" detector_config.device = "GPU"
if not os.path.isfile(detector_config.model.path):
logger.error(f"OpenVino model file {detector_config.model.path} not found.")
raise FileNotFoundError
self.interpreter = self.ov_core.compile_model( self.interpreter = self.ov_core.compile_model(
model=detector_config.model.path, device_name=detector_config.device model=detector_config.model.path, device_name=detector_config.device
) )