diff --git a/frigate/detectors/plugins/tensorrt.py b/frigate/detectors/plugins/tensorrt.py index 0f41d7a8a..d9006fe86 100644 --- a/frigate/detectors/plugins/tensorrt.py +++ b/frigate/detectors/plugins/tensorrt.py @@ -175,9 +175,6 @@ class TensorRtDetector(DetectionApi): if len(binding_dims) == 4: # explicit batch case (TensorRT 7+) 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: raise ValueError( "bad dims of binding %s: %s" % (binding, str(binding_dims))