apply conf th to post process

2026-02-09 04:35:25 +03:00 · 2024-02-21 02:01:24 -06:00 · 2024-02-21 02:01:24 -06:00 · 95c04b6244
commit 95c04b6244
parent adfaa04760
1 changed files with 3 additions and 1 deletions
--- 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)