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

Formatting

Browse Source
This commit is contained in:
Nicolas Mowen 2025-09-14 07:03:23 -06:00
parent 34d7bc9e6c
commit 05e5d9709c
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
frigate/detectors/detection_runners.py
View File

@ -186,15 +186,15 @@ class OpenVINOModelRunner(BaseModelRunner):
if port.get_any_name() == input_name: if port.get_any_name() == input_name:
input_port = port input_port = port
break break
if input_port is None: if input_port is None:
raise ValueError(f"Input '{input_name}' not found in model") raise ValueError(f"Input '{input_name}' not found in model")
# Create tensor with the correct element type # Create tensor with the correct element type
input_element_type = input_port.get_element_type() input_element_type = input_port.get_element_type()
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 # Set the input tensor
self.infer_request.set_input_tensor(input_tensor) self.infer_request.set_input_tensor(input_tensor)