From 95c04b624421f0f792ab9c861500a1d9173877dc Mon Sep 17 00:00:00 2001 From: alexyao2015 <33379584+alexyao2015@users.noreply.github.com> Date: Wed, 21 Feb 2024 02:01:24 -0600 Subject: [PATCH] apply conf th to post process --- frigate/detectors/plugins/tensorrt.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) 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)