administrator/frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2026-08-07 03:11:15 +03:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

Improve Intel Model (#20541)

Browse Source 
* Update supported models and inference times

* Fix d-fine inputs

* Improve d-fine
This commit is contained in:
Nicolas Mowen
2025-10-17 06:31:28 -06:00
committed by GitHub
parent d7275a3c1a
commit 4228861810
3 changed files with 11 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files
frigate/detectors/detection_runners.py
+6 -4
View File
@@ -312,11 +312,13 @@ class OpenVINOModelRunner(BaseModelRunner):
# Multiple inputs case - set each input by name
for input_name, input_data in inputs.items():
# Find the input by name
# Find the input by name and its index
input_port = None
for port in self.compiled_model.inputs:
input_index = None
for idx, port in enumerate(self.compiled_model.inputs):
if port.get_any_name() == input_name:
input_port = port
input_index = idx
break
if input_port is None:
@@ -327,8 +329,8 @@ class OpenVINOModelRunner(BaseModelRunner):
input_tensor = ov.Tensor(input_element_type, input_data.shape)
np.copyto(input_tensor.data, input_data)
# Set the input tensor
self.infer_request.set_input_tensor(input_tensor)
# Set the input tensor for the specific port index
self.infer_request.set_input_tensor(input_index, input_tensor)
# Run inference
self.infer_request.infer()