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

Improve d-fine

Browse Source
This commit is contained in:
Nicolas Mowen 2025-10-17 06:19:57 -06:00
parent facd0f287d
commit 7db9a9c07e
1 changed files with 6 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
frigate/detectors/detection_runners.py
View File

@ -312,11 +312,13 @@ class OpenVINOModelRunner(BaseModelRunner):
# Multiple inputs case - set each input by name # Multiple inputs case - set each input by name
for input_name, input_data in inputs.items(): for input_name, input_data in inputs.items():
# Find the input by name # Find the input by name and its index
input_port = None 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: if port.get_any_name() == input_name:
input_port = port input_port = port
input_index = idx
break break
if input_port is None: if input_port is None:
@ -327,8 +329,8 @@ class OpenVINOModelRunner(BaseModelRunner):
input_tensor = ov.Tensor(input_element_type, input_data.shape) input_tensor = ov.Tensor(input_element_type, input_data.shape)
np.copyto(input_tensor.data, input_data) np.copyto(input_tensor.data, input_data)
# Set the input tensor for the specific port # Set the input tensor for the specific port index
self.infer_request.set_input_tensor(input_port, input_tensor) self.infer_request.set_input_tensor(input_index, input_tensor)
# Run inference # Run inference
self.infer_request.infer() self.infer_request.infer()