diff --git a/frigate/detectors/plugins/tensorrt.py b/frigate/detectors/plugins/tensorrt.py index c7653b159..f1f91da0c 100644 --- a/frigate/detectors/plugins/tensorrt.py +++ b/frigate/detectors/plugins/tensorrt.py @@ -313,7 +313,7 @@ class TensorRtDetector(DetectionApi): ordered[:, 3] = np.clip(ordered[:, 3] + ordered[:, 1], 0, 1) # put result into the correct order and limit to top 20 detections = ordered[:, [5, 4, 1, 0, 3, 2]][:20] - + # pad to 20x6 shape append_cnt = 20 - len(detections) if append_cnt > 0: