From f29e152619173834d808afd514ccc40668f5db26 Mon Sep 17 00:00:00 2001 From: Nicolas Mowen Date: Tue, 7 Nov 2023 04:32:52 -0700 Subject: [PATCH] Revert TRT class id filtering (#8497) --- frigate/detectors/plugins/tensorrt.py | 10 ---------- 1 file changed, 10 deletions(-) diff --git a/frigate/detectors/plugins/tensorrt.py b/frigate/detectors/plugins/tensorrt.py index 1ca67b683..2a57ec2d3 100644 --- a/frigate/detectors/plugins/tensorrt.py +++ b/frigate/detectors/plugins/tensorrt.py @@ -293,16 +293,6 @@ class TensorRtDetector(DetectionApi): # raw_detections: Nx7 numpy arrays of # [[x, y, w, h, box_confidence, class_id, class_prob], - # throw out any detections with negative class IDs - valid_detections = [] - for r in raw_detections: - if r[5] >= 0: - valid_detections.append(r) - else: - logger.warning(f"Found TensorRT detection with invalid class id {r}") - - raw_detections = valid_detections - # Calculate score as box_confidence x class_prob raw_detections[:, 4] = raw_detections[:, 4] * raw_detections[:, 6] # Reorder elements by the score, best on top, remove class_prob