diff --git a/frigate/detectors/plugins/tensorrt.py b/frigate/detectors/plugins/tensorrt.py index 0c49056d4..c0967e724 100644 --- a/frigate/detectors/plugins/tensorrt.py +++ b/frigate/detectors/plugins/tensorrt.py @@ -106,8 +106,8 @@ class TensorRtDetector(DetectionApi): for i in range(self.engine.num_bindings): name = self.engine.get_tensor_name(i) shape = ( - tuple(self.engine.get_binding_shape(name)), - trt.nptype(self.engine.get_binding_dtype(name)), + tuple(self.engine.get_tensor_shape(name)), + trt.nptype(self.engine.get_tensor_dtype(name)), ) if self.engine.get_tensor_mode(name) == trt.TensorIOMode.INPUT: input_shape = shape