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

Correctly calculate input data type for OV (#20066)

Browse Source 
* Correctly calculate input data type for OV

* Formatting
This commit is contained in:
Nicolas Mowen
2025-09-14 07:15:41 -06:00
committed by GitHub
parent c74e86dff2
commit ff0430964c
2 changed files with 38 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files
frigate/detectors/plugins/openvino.py
+6 -5
View File
@@ -141,16 +141,17 @@ class OvDetector(DetectionApi):
"images": tensor_input,
"orig_target_sizes": np.array([[self.h, self.w]], dtype=np.int64),
}
outputs = self.runner.run_with_named_inputs(inputs)
outputs = self.runner.run(inputs)
tensor_output = (
outputs["output0"],
outputs["output1"],
outputs["output2"],
outputs[0],
outputs[1],
outputs[2],
)
return post_process_dfine(tensor_output, self.w, self.h)
# Run inference using the runner
outputs = self.runner.run(tensor_input)
input_name = self.runner.get_input_names()[0]
outputs = self.runner.run({input_name: tensor_input})
detections = np.zeros((20, 6), np.float32)