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