mirror of
https://github.com/blakeblackshear/frigate.git
synced 2026-04-23 23:34:32 +03:00
Fix
This commit is contained in:
parent
d1f4e73fe4
commit
adc786e41f
@ -190,7 +190,7 @@ class OpenVINOModelRunner(BaseModelRunner):
|
|||||||
self.input_tensor = ov.Tensor(input_element_type, input_shape)
|
self.input_tensor = ov.Tensor(input_element_type, input_shape)
|
||||||
except RuntimeError:
|
except RuntimeError:
|
||||||
# model is complex and has dynamic shape
|
# model is complex and has dynamic shape
|
||||||
raise
|
pass
|
||||||
|
|
||||||
def get_input_names(self) -> list[str]:
|
def get_input_names(self) -> list[str]:
|
||||||
"""Get input names for the model."""
|
"""Get input names for the model."""
|
||||||
|
|||||||
@ -43,7 +43,7 @@ class OvDetector(DetectionApi):
|
|||||||
self.w = detector_config.model.width
|
self.w = detector_config.model.width
|
||||||
|
|
||||||
self.runner = OpenVINOModelRunner(
|
self.runner = OpenVINOModelRunner(
|
||||||
model_path=detector_config.model.path, device=detector_config.device
|
model_path=detector_config.model.path, device=detector_config.device, complex_model=False
|
||||||
)
|
)
|
||||||
|
|
||||||
# For dfine models, also pre-allocate target sizes tensor
|
# For dfine models, also pre-allocate target sizes tensor
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user