diff --git a/frigate/detectors/plugins/tensorrt.py b/frigate/detectors/plugins/tensorrt.py index 50f6e036c..8bc4aac98 100644 --- a/frigate/detectors/plugins/tensorrt.py +++ b/frigate/detectors/plugins/tensorrt.py @@ -300,7 +300,9 @@ class TensorRtDetector(DetectionApi): for o in trt_outputs: detections.append( yolov8_postprocess( - self.input_shape[0], o.reshape(self.output_shape[0]) + self.input_shape[0], + o.reshape(self.output_shape[0]), + score_threshold=self.conf_th, ), ) detections = np.concatenate(detections)