From 13f9fe3b486180937fce2172be04a99a7d98573f Mon Sep 17 00:00:00 2001 From: alexyao2015 <33379584+alexyao2015@users.noreply.github.com> Date: Wed, 21 Feb 2024 01:36:54 -0600 Subject: [PATCH] use tensor api --- frigate/detectors/plugins/tensorrt.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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