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

Remove unnecessary size check in TensorRT v10 block

Browse Source
This commit is contained in:
Rémi Bédard-Couture 2024-05-05 10:01:35 -04:00
parent 9a642086f9
commit 408f295416
1 changed files with 0 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -175,9 +175,6 @@ class TensorRtDetector(DetectionApi):
if len(binding_dims) == 4: if len(binding_dims) == 4:
# explicit batch case (TensorRT 7+) # explicit batch case (TensorRT 7+)
size = trt.volume(binding_dims) size = trt.volume(binding_dims)
elif len(binding_dims) == 3:
# implicit batch case (TensorRT 6 or older)
size = trt.volume(binding_dims) * self.engine.max_batch_size
else: else:
raise ValueError( raise ValueError(
"bad dims of binding %s: %s" % (binding, str(binding_dims)) "bad dims of binding %s: %s" % (binding, str(binding_dims))