From 485f3075747a9754223280531749f3fbfc69401d Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?R=C3=A9mi=20B=C3=A9dard-Couture?= Date: Tue, 7 May 2024 22:38:07 -0400 Subject: [PATCH] Fix wrong function name in new _get_binding_dtype function and only return input check (not assertion) in new _binding_is_input function --- frigate/detectors/plugins/tensorrt.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/frigate/detectors/plugins/tensorrt.py b/frigate/detectors/plugins/tensorrt.py index b3bcd360c..d43b9c902 100644 --- a/frigate/detectors/plugins/tensorrt.py +++ b/frigate/detectors/plugins/tensorrt.py @@ -91,10 +91,9 @@ class TensorRtDetector(DetectionApi): def _binding_is_input(self, binding): if TRT_VERSION < 10: - assert self.engine.binding_is_input(binding) + return self.engine.binding_is_input(binding) else: - assert binding == "input" - return True + return binding == "input" def _get_binding_dims(self, binding): if TRT_VERSION < 10: @@ -106,7 +105,7 @@ class TensorRtDetector(DetectionApi): if TRT_VERSION < 10: return self.engine.get_binding_dtype(binding) else: - return self.engine.get_tensor_shape(binding) + return self.engine.get_tensor_dtype(binding) def _execute(self): if TRT_VERSION < 10: