mirror of
https://github.com/blakeblackshear/frigate.git
synced 2026-04-15 03:22:10 +03:00
Formatting
This commit is contained in:
parent
34d7bc9e6c
commit
05e5d9709c
@ -186,15 +186,15 @@ class OpenVINOModelRunner(BaseModelRunner):
|
||||
if port.get_any_name() == input_name:
|
||||
input_port = port
|
||||
break
|
||||
|
||||
|
||||
if input_port is None:
|
||||
raise ValueError(f"Input '{input_name}' not found in model")
|
||||
|
||||
|
||||
# Create tensor with the correct element type
|
||||
input_element_type = input_port.get_element_type()
|
||||
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)
|
||||
|
||||
|
||||
Loading…
Reference in New Issue
Block a user