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

Cleanup

Browse Source
This commit is contained in:
Nicolas Mowen 2025-09-17 15:10:02 -06:00
parent adc786e41f
commit fd0aa5e654
2 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
frigate/detectors/detection_runners.py
View File

@ -391,7 +391,9 @@ def get_optimized_runner(
return RKNNModelRunner(rknn_path)
if device != "CPU" and is_openvino_gpu_npu_available():
return OpenVINOModelRunner(model_path, device or "AUTO", complex_model, **kwargs)
return OpenVINOModelRunner(
model_path, device or "AUTO", complex_model, **kwargs
)
providers, options = get_ort_providers(device == "CPU", device, **kwargs)
ortSession = ort.InferenceSession(

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

@ -43,7 +43,9 @@ class OvDetector(DetectionApi):
self.w = detector_config.model.width
self.runner = OpenVINOModelRunner(
model_path=detector_config.model.path, device=detector_config.device, complex_model=False
model_path=detector_config.model.path,
device=detector_config.device,
complex_model=False,
)
# For dfine models, also pre-allocate target sizes tensor