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

apply conf th to post process

Browse Source
This commit is contained in:
alexyao2015 2024-02-21 02:01:24 -06:00
parent adfaa04760
commit 95c04b6244
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -300,7 +300,9 @@ class TensorRtDetector(DetectionApi):
for o in trt_outputs: for o in trt_outputs:
detections.append( detections.append(
yolov8_postprocess( 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) detections = np.concatenate(detections)