diff --git a/frigate/detectors/detection_api.py b/frigate/detectors/detection_api.py index 6e1224272..c27e64362 100644 --- a/frigate/detectors/detection_api.py +++ b/frigate/detectors/detection_api.py @@ -22,7 +22,7 @@ class DetectionApi(ABC): def detect_raw(self, tensor_input): pass - def yolonas(self, output): + def post_process_yolonas(self, output): """ @param output: output of inference expected shape: [np.array(1, N, 4), np.array(1, N, 80)]