From 459f87995bf8d8c08b3c9f8b70146b410a9faa41 Mon Sep 17 00:00:00 2001 From: Nicolas Mowen Date: Fri, 21 Mar 2025 18:46:27 -0600 Subject: [PATCH] Cleanup --- frigate/detectors/plugins/onnx.py | 2 +- frigate/util/model.py | 1 - 2 files changed, 1 insertion(+), 2 deletions(-) diff --git a/frigate/detectors/plugins/onnx.py b/frigate/detectors/plugins/onnx.py index 03940f418..2679185a9 100644 --- a/frigate/detectors/plugins/onnx.py +++ b/frigate/detectors/plugins/onnx.py @@ -11,8 +11,8 @@ from frigate.detectors.detector_config import ( ) from frigate.util.model import ( get_ort_providers, - post_process_rfdetr, post_process_dfine, + post_process_rfdetr, post_process_yolov9, ) diff --git a/frigate/util/model.py b/frigate/util/model.py index 55a5de86c..19b3b1bf5 100644 --- a/frigate/util/model.py +++ b/frigate/util/model.py @@ -95,7 +95,6 @@ def post_process_rfdetr(tensor_output: list[np.ndarray, np.ndarray]) -> np.ndarr bbox[3], bbox[2], ] - # print(f"found a detection {detections[i]}") return detections