administrator/frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2026-02-11 13:45:25 +03:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

Fix wrong function name in new _get_binding_dtype function and only return input check (not assertion) in new _binding_is_input function

Browse Source
This commit is contained in:
Rémi Bédard-Couture 2024-05-07 22:38:07 -04:00
parent 6c0abe4833
commit 485f307574
1 changed files with 3 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
frigate/detectors/plugins/tensorrt.py
View File

@ -91,10 +91,9 @@ class TensorRtDetector(DetectionApi):
def _binding_is_input(self, binding): def _binding_is_input(self, binding):
if TRT_VERSION < 10: if TRT_VERSION < 10:
assert self.engine.binding_is_input(binding) return self.engine.binding_is_input(binding)
else: else:
assert binding == "input" return binding == "input"
return True
def _get_binding_dims(self, binding): def _get_binding_dims(self, binding):
if TRT_VERSION < 10: if TRT_VERSION < 10:
@ -106,7 +105,7 @@ class TensorRtDetector(DetectionApi):
if TRT_VERSION < 10: if TRT_VERSION < 10:
return self.engine.get_binding_dtype(binding) return self.engine.get_binding_dtype(binding)
else: else:
return self.engine.get_tensor_shape(binding) return self.engine.get_tensor_dtype(binding)
def _execute(self): def _execute(self):
if TRT_VERSION < 10: if TRT_VERSION < 10: